Atomic-Ai
/

AtomicGPT-T1-18042025

Model card Files Files and versions

Loewolf commited on Apr 20

Commit

261ec82

·

verified ·

1 Parent(s): 736afcc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -107,7 +107,7 @@ chatbot = pipeline(
 def generate_response(prompt):
     output = chatbot(
         prompt,
-        max_new_tokens=1024,
         eos_token_id=50268,
         pad_token_id=tokenizer.pad_token_id,
         do_sample=True,

 def generate_response(prompt):
     output = chatbot(
         prompt,
+        max_new_tokens=2048,
         eos_token_id=50268,
         pad_token_id=tokenizer.pad_token_id,
         do_sample=True,