srinivasbilla commited on
Commit
e007da5
·
verified ·
1 Parent(s): 2779b9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -14,10 +14,7 @@ tokenizer = AutoTokenizer.from_pretrained(llasa_3b)
14
  model = AutoModelForCausalLM.from_pretrained(
15
  llasa_3b,
16
  trust_remote_code=True,
17
- use_cache=False,
18
- torch_dtype=torch.bfloat16,
19
  device_map='cuda',
20
- return_dict=True
21
  )
22
 
23
  model_path = "srinivasbilla/xcodec2"
 
14
  model = AutoModelForCausalLM.from_pretrained(
15
  llasa_3b,
16
  trust_remote_code=True,
 
 
17
  device_map='cuda',
 
18
  )
19
 
20
  model_path = "srinivasbilla/xcodec2"