Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -176,7 +176,7 @@ def process_translate(language: str, audio_path: str) -> str:
|
|
| 176 |
inputs = processor.apply_chat_template(conversation)
|
| 177 |
print("OK")
|
| 178 |
inputs = inputs.to(device, dtype=torch.bfloat16)
|
| 179 |
-
print(
|
| 180 |
outputs = model.generate(**inputs, max_new_tokens=MAX_TOKENS)
|
| 181 |
print("OK")
|
| 182 |
print("processor.batch_decode(")
|
|
|
|
| 176 |
inputs = processor.apply_chat_template(conversation)
|
| 177 |
print("OK")
|
| 178 |
inputs = inputs.to(device, dtype=torch.bfloat16)
|
| 179 |
+
print("model.generate(**inputs, max_new_tokens=MAX_TOKENS) ...")
|
| 180 |
outputs = model.generate(**inputs, max_new_tokens=MAX_TOKENS)
|
| 181 |
print("OK")
|
| 182 |
print("processor.batch_decode(")
|