Hugging Face's logo

Spaces:
han1997
/
cobra
Runtime error

App Files Community
3
han1997 commited on
Commit
607c954
·
verified ·
1 Parent(s): e8f28cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -31,6 +31,7 @@ else:
31
  DEVICE = "cpu"
32
  DTYPE = torch.float32
33
  vlm.to(DEVICE, dtype=DTYPE)
 
34
 
35
  prompt_builder = vlm.get_prompt_builder()
36
 
 
31
  DEVICE = "cpu"
32
  DTYPE = torch.float32
33
  vlm.to(DEVICE, dtype=DTYPE)
34
+ vlm.llm_backbone.half_precision_dtype = DTYPE
35
 
36
  prompt_builder = vlm.get_prompt_builder()
37