AlexWortega
/

wortegaLM

Text Generation

Inference Endpoints

Model card Files Files and versions Community

AlexWortega commited on May 3, 2023

Commit

74a13a9

•

1 Parent(s): 214d407

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -68,7 +68,7 @@ def generate_seqs(q,model, k=2):
         "num_beams": 4,
         "num_return_sequences": k
     }
-    q = q
     t = tokenizer.encode(q, add_special_tokens=False, return_tensors='pt').to(device)
     g = model.generate(t, **gen_kwargs)
     generated_sequences = tokenizer.batch_decode(g, skip_special_tokens=False)

         "num_beams": 4,
         "num_return_sequences": k
     }
     t = tokenizer.encode(q, add_special_tokens=False, return_tensors='pt').to(device)
     g = model.generate(t, **gen_kwargs)
     generated_sequences = tokenizer.batch_decode(g, skip_special_tokens=False)