Hugging Face's logo

Spaces:
han1997
/
cobra
Runtime error

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,8 +30,8 @@ if torch.cuda.is_available():
30
  else:
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
 
 
30
  else:
31
  DEVICE = "cpu"
32
  DTYPE = torch.float32
33
+ vlm.enable_mixed_precision_training = False
34
  vlm.to(DEVICE, dtype=DTYPE)
 
35
 
36
  prompt_builder = vlm.get_prompt_builder()
37