Spaces:
Runtime error
Runtime error
John Ho commited on
Commit ·
4dc5aed
1
Parent(s): c5ab6e1
debugging internvl3
Browse files
app.py
CHANGED
|
@@ -214,6 +214,7 @@ def inference(
|
|
| 214 |
).to("cuda", dtype=DTYPE)
|
| 215 |
|
| 216 |
output = model.generate(**inputs, max_new_tokens=max_tokens)
|
|
|
|
| 217 |
output_text = processor.decode(
|
| 218 |
output[0, inputs["input_ids"].shape[1] :], skip_special_tokens=True
|
| 219 |
)
|
|
|
|
| 214 |
).to("cuda", dtype=DTYPE)
|
| 215 |
|
| 216 |
output = model.generate(**inputs, max_new_tokens=max_tokens)
|
| 217 |
+
logger.debug(output)
|
| 218 |
output_text = processor.decode(
|
| 219 |
output[0, inputs["input_ids"].shape[1] :], skip_special_tokens=True
|
| 220 |
)
|