Spaces:
Runtime error
Runtime error
zero embeddings
Browse files
app_v2.py
CHANGED
|
@@ -52,7 +52,7 @@ def bot(history):
|
|
| 52 |
except IOError:
|
| 53 |
response = "Failed to open image. Please check the image path or file permissions."
|
| 54 |
else:
|
| 55 |
-
image_embeds = torch.zeros(1, 729, 2048)
|
| 56 |
response = moondream.answer_question(image_embeds, last_message, tokenizer)
|
| 57 |
|
| 58 |
history[-1][1] = ""
|
|
|
|
| 52 |
except IOError:
|
| 53 |
response = "Failed to open image. Please check the image path or file permissions."
|
| 54 |
else:
|
| 55 |
+
image_embeds = torch.zeros(1, 729, 2048, dtype=torch.bfloat16, device='cuda')
|
| 56 |
response = moondream.answer_question(image_embeds, last_message, tokenizer)
|
| 57 |
|
| 58 |
history[-1][1] = ""
|