Update app.py
Browse files
app.py
CHANGED
|
@@ -31,8 +31,6 @@ def generate(your_prompt, max_new_tokens, repetition_penalty, temperature, model
|
|
| 31 |
torch.manual_seed(seed)
|
| 32 |
else:
|
| 33 |
torch.manual_seed(seed)
|
| 34 |
-
|
| 35 |
-
print(f"seed has been set to {seed}")
|
| 36 |
|
| 37 |
outputs = model.generate(
|
| 38 |
input_ids,
|
|
|
|
| 31 |
torch.manual_seed(seed)
|
| 32 |
else:
|
| 33 |
torch.manual_seed(seed)
|
|
|
|
|
|
|
| 34 |
|
| 35 |
outputs = model.generate(
|
| 36 |
input_ids,
|