wasmdashai
commited on
Commit
•
169c2a7
1
Parent(s):
b727876
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ token=os.environ.get("key_")
|
|
12 |
ckpt = "meta-llama/Llama-3.2-11B-Vision-Instruct"
|
13 |
model = MllamaForConditionalGeneration.from_pretrained(ckpt,token=token,
|
14 |
torch_dtype=torch.bfloat16)#.to("cuda")
|
15 |
-
processor = AutoProcessor.from_pretrained(ckpt)
|
16 |
|
17 |
|
18 |
#@spaces.GPU
|
|
|
12 |
ckpt = "meta-llama/Llama-3.2-11B-Vision-Instruct"
|
13 |
model = MllamaForConditionalGeneration.from_pretrained(ckpt,token=token,
|
14 |
torch_dtype=torch.bfloat16)#.to("cuda")
|
15 |
+
processor = AutoProcessor.from_pretrained(ckpt,token=token)
|
16 |
|
17 |
|
18 |
#@spaces.GPU
|