Update Main.py
Browse files
Main.py
CHANGED
|
@@ -126,7 +126,7 @@ inputs = {
|
|
| 126 |
}
|
| 127 |
|
| 128 |
# Генерация
|
| 129 |
-
|
| 130 |
**inputs,
|
| 131 |
max_new_tokens=2048,
|
| 132 |
do_sample=False,
|
|
|
|
| 126 |
}
|
| 127 |
|
| 128 |
# Генерация
|
| 129 |
+
output_ids = model.generate(
|
| 130 |
**inputs,
|
| 131 |
max_new_tokens=2048,
|
| 132 |
do_sample=False,
|