Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ model_name = 'AIDC-AI/Ovis1.6-Llama3.2-3B'
|
|
14 |
model = AutoModelForCausalLM.from_pretrained(model_name,
|
15 |
torch_dtype=torch.bfloat16,
|
16 |
multimodal_max_length=8192,
|
17 |
-
|
18 |
trust_remote_code=True).to(device='cuda')
|
19 |
text_tokenizer = model.get_text_tokenizer()
|
20 |
visual_tokenizer = model.get_visual_tokenizer()
|
|
|
14 |
model = AutoModelForCausalLM.from_pretrained(model_name,
|
15 |
torch_dtype=torch.bfloat16,
|
16 |
multimodal_max_length=8192,
|
17 |
+
token=os.getenv('HUGGINGFACE_TOKEN'),
|
18 |
trust_remote_code=True).to(device='cuda')
|
19 |
text_tokenizer = model.get_text_tokenizer()
|
20 |
visual_tokenizer = model.get_visual_tokenizer()
|