AliMuhammad73 commited on
Commit
f960f9b
·
verified ·
1 Parent(s): 851fe17

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -56,7 +56,7 @@ inputs = tokenizer.encode(prompt_urdu)
56
  inputs_tensor = torch.tensor(inputs).unsqueeze(0) # Add batch dimension
57
 
58
  # Generate text
59
- outputs = model.generate(inputs_tensor, max_new_tokens=128, temperature=0.7)
60
  outputs_tensor = torch.tensor(outputs).unsqueeze(0)
61
  generated_text = tokenizer.decode(outputs_tensor[0].squeeze().tolist())
62
 
 
56
  inputs_tensor = torch.tensor(inputs).unsqueeze(0) # Add batch dimension
57
 
58
  # Generate text
59
+ outputs = model.generate(inputs_tensor, max_new_tokens=64, temperature=0.7)
60
  outputs_tensor = torch.tensor(outputs).unsqueeze(0)
61
  generated_text = tokenizer.decode(outputs_tensor[0].squeeze().tolist())
62