Hugging Face's logo

Spaces:
kkr5155
/
maya_demo
Running on Zero

App Files Community
2
kkr5155 commited on
Commit
606ed65
·
verified ·
1 Parent(s): b861a1a

Update eval_utils.py

Browse files
Files changed (1) hide show
  1. eval_utils.py +1 -1
eval_utils.py CHANGED
@@ -48,7 +48,7 @@ def load_maya_model(model_base: str, model_path : str, projector_path : Optional
48
 
49
  device_map = 'auto'
50
  kwargs = {"device_map": device_map}
51
- kwargs['torch_dtype'] = torch.float16
52
  # kwargs['attn_implementation'] = 'flash_attention_2'
53
 
54
  ## Instantiating tokenizer and model base
 
48
 
49
  device_map = 'auto'
50
  kwargs = {"device_map": device_map}
51
+ kwargs['torch_dtype'] = torch.float32
52
  # kwargs['attn_implementation'] = 'flash_attention_2'
53
 
54
  ## Instantiating tokenizer and model base