sag-uniroma2
/

extremITA-Camoscio-7b

Text Generation

text-generation-inference

Model card Files Files and versions Community

cdh commited on Aug 24, 2023

Commit

b15a64c

•

1 Parent(s): ba51b62

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -39,7 +39,8 @@ prompts = [
   stampa 'FizzBuzz'."
 ]
-inputs = tokenizer(prompts, return_tensors="pt", padding=True, truncation=True).to(model.device)
 with torch.no_grad():
   gen_outputs = model.generate(

   stampa 'FizzBuzz'."
 ]
+inputs = tokenizer(prompts, return_tensors="pt", padding=True, \
+                  truncation=True).to(model.device)
 with torch.no_grad():
   gen_outputs = model.generate(