Spaces:

joaoargolo1
/

emollama

Runtime error

joaoargolo1 commited on May 2

Commit

384993b

verified ·

1 Parent(s): 77a58f0

Update models.py

Files changed (1) hide show

models.py CHANGED Viewed

@@ -21,7 +21,6 @@ Assistente:"""
     inputs = tokenizer(prompt, return_tensors="pt", padding=True).to(model.device)
-    # Geração com attention_mask explícita
     with torch.no_grad():
         generate_ids = model.generate(
             inputs["input_ids"],

     inputs = tokenizer(prompt, return_tensors="pt", padding=True).to(model.device)
     with torch.no_grad():
         generate_ids = model.generate(
             inputs["input_ids"],