Spaces:
Runtime error
Runtime error
Commit
•
e77f167
1
Parent(s):
b67ea1d
Update app.py
Browse files
app.py
CHANGED
|
@@ -136,7 +136,7 @@ images_tensor = process_images(
|
|
| 136 |
input_ids = (
|
| 137 |
tokenizer_image_token(prompt, tokenizer, IMAGE_TOKEN_INDEX, return_tensors="pt")
|
| 138 |
.unsqueeze(0)
|
| 139 |
-
.cuda()
|
| 140 |
)
|
| 141 |
|
| 142 |
with torch.inference_mode():
|
|
|
|
| 136 |
input_ids = (
|
| 137 |
tokenizer_image_token(prompt, tokenizer, IMAGE_TOKEN_INDEX, return_tensors="pt")
|
| 138 |
.unsqueeze(0)
|
| 139 |
+
# .cuda()
|
| 140 |
)
|
| 141 |
|
| 142 |
with torch.inference_mode():
|