sitammeur commited on
Commit
77bcd66
·
verified ·
1 Parent(s): df12f2d

Update src/app/model.py

Browse files
Files changed (1) hide show
  1. src/app/model.py +3 -0
src/app/model.py CHANGED
@@ -38,6 +38,9 @@ def load_model_tokenizer_and_processor(model_name: str, device: str) -> Any:
38
  trust_remote_code=True,
39
  attn_implementation="sdpa",
40
  torch_dtype=torch.bfloat16,
 
 
 
41
  token=access_token
42
  )
43
  model = model.eval().to(device=device)
 
38
  trust_remote_code=True,
39
  attn_implementation="sdpa",
40
  torch_dtype=torch.bfloat16,
41
+ init_vision=True,
42
+ init_audio=False,
43
+ init_tts=False,
44
  token=access_token
45
  )
46
  model = model.eval().to(device=device)