Spaces:
Running
on
Zero
Running
on
Zero
initialization
Browse files
README.md
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
---
|
2 |
-
title: DoctorXRay (XRayAnalysis V1.
|
3 |
emoji: 🐠
|
4 |
colorFrom: pink
|
5 |
colorTo: purple
|
|
|
1 |
---
|
2 |
+
title: DoctorXRay (XRayAnalysis V1.2)
|
3 |
emoji: 🐠
|
4 |
colorFrom: pink
|
5 |
colorTo: purple
|
app.py
CHANGED
@@ -143,7 +143,7 @@ def load_model():
|
|
143 |
# torch_dtype=torch.float16
|
144 |
# ).to('cuda')
|
145 |
model = AutoModelForVision2Seq.from_pretrained(
|
146 |
-
"Aekanun/Llama-3.2-11B-Vision-Instruct-XRay",
|
147 |
load_in_4bit = True,
|
148 |
device_map="auto",
|
149 |
torch_dtype = torch.float16
|
|
|
143 |
# torch_dtype=torch.float16
|
144 |
# ).to('cuda')
|
145 |
model = AutoModelForVision2Seq.from_pretrained(
|
146 |
+
"Aekanun/Llama-3.2-11B-Vision-Instruct-XRay-v1.0",
|
147 |
load_in_4bit = True,
|
148 |
device_map="auto",
|
149 |
torch_dtype = torch.float16
|