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] = ""
|