File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -655,7 +655,7 @@ def generate(
655
655
assert False , "Unexpected type in prompt"
656
656
657
657
if gen_settings is None :
658
- p_settings = ExLlamaV2Sampler . Settings ()
658
+ p_settings = None
659
659
elif isinstance (gen_settings , ExLlamaV2Sampler .Settings ):
660
660
p_settings = gen_settings
661
661
elif isinstance (gen_settings , list ):
@@ -1536,7 +1536,7 @@ def __init__(
1536
1536
max_new_tokens : int ,
1537
1537
min_new_tokens : int = 0 ,
1538
1538
max_skips : int | None = 4 ,
1539
- gen_settings : ExLlamaV2Sampler .Settings = ExLlamaV2Sampler . Settings () ,
1539
+ gen_settings : ExLlamaV2Sampler .Settings | None = None ,
1540
1540
seed : int = None ,
1541
1541
stop_conditions : list | tuple | set = None ,
1542
1542
decode_special_tokens : bool = False ,
@@ -1629,6 +1629,9 @@ def __init__(
1629
1629
self .max_skips = max_skips
1630
1630
self .allocated_pages = None
1631
1631
1632
+ if gen_settings is None :
1633
+ gen_settings = ExLlamaV2Sampler .Settings ()
1634
+
1632
1635
# Prepare sequences
1633
1636
1634
1637
if not isinstance (input_ids , list ):
You can’t perform that action at this time.
0 commit comments