Spaces:
Running
on
Zero
Running
on
Zero
srinivasbilla
commited on
Update app.py
Browse files
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"
|