Update app.py
Browse files
app.py
CHANGED
@@ -31,6 +31,8 @@ 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 |
outputs = model.generate(
|
36 |
input_ids,
|
|
|
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,
|