Rustamshry
/

Qwen2.5-1.5B-Medical-Az

Question Answering

Model card Files Files and versions

Rustamshry commited on May 11

Commit

165a185

·

verified ·

1 Parent(s): 41de2fe

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -74,7 +74,7 @@ input_ids = tokenizer(prompt, return_tensors="pt").to(model.device)
 outputs = model.generate(
     **input_ids,
-    max_new_tokens=1024,
     #temperature=0.7,
     #top_p=0.9,
     #do_sample=True,

 outputs = model.generate(
     **input_ids,
+    max_new_tokens=2000,
     #temperature=0.7,
     #top_p=0.9,
     #do_sample=True,