sultan-hassan
/

CosmoGemma_2b_en

Text Generation

Model card Files Files and versions Community

sultan-hassan commited on 30 days ago

Commit

6b76d6d

•

1 Parent(s): 9a2225d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -48,7 +48,7 @@ prompt = template.format(
   response="",
   )
 out = gemma_lm.generate(prompt, max_length=1024)
-ind = out.index('Response') + len('Response')+2
 print ("Question:", Question)
 print ("Answer:", out[ind:])
 ```

   response="",
   )
 out = gemma_lm.generate(prompt, max_length=1024)
+ind = out.index('\n\nResponse:\n') + len('\n\nResponse:\n')
 print ("Question:", Question)
 print ("Answer:", out[ind:])
 ```