hanzla commited on
Commit
2bf6cea
1 Parent(s): 104d077

zero embeddings

Browse files
Files changed (1) hide show
  1. app_v2.py +1 -1
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] = ""