Crystalcareai
/

Quiet-Star-Custom

Text Generation

Model card Files Files and versions Community

Crystalcareai commited on Apr 6, 2024

Commit

6356c8a

·

verified ·

1 Parent(s): 257e2f1

Update generate.py

Files changed (1) hide show

generate.py +1 -1

generate.py CHANGED Viewed

@@ -172,7 +172,7 @@ def generate(
     generated_token_ids = custom_generate(
         self,
-        input_ids=input_ids,
         attention_mask=attention_mask,
         max_new_tokens=max_new_tokens,
         min_length=min_length,

     generated_token_ids = custom_generate(
         self,
+        input_ids,
         attention_mask=attention_mask,
         max_new_tokens=max_new_tokens,
         min_length=min_length,