Spaces:

Aekanun
/

DocterXray-v1.2

Runtime error

Aekanun commited on Nov 26, 2024

Commit

7c418c1

1 Parent(s): 34624f8

initialization

Files changed (2) hide show

README.md CHANGED Viewed

@@ -1,5 +1,5 @@
 ---
-title: DoctorXRay (XRayAnalysis V1.1)
 emoji: 🐠
 colorFrom: pink
 colorTo: purple

 ---
+title: DoctorXRay (XRayAnalysis V1.2)
 emoji: 🐠
 colorFrom: pink
 colorTo: purple

app.py CHANGED Viewed

@@ -143,7 +143,7 @@ def load_model():
         #     torch_dtype=torch.float16
         # ).to('cuda')
         model = AutoModelForVision2Seq.from_pretrained(
-            "Aekanun/Llama-3.2-11B-Vision-Instruct-XRay",
             load_in_4bit = True,
             device_map="auto",
             torch_dtype = torch.float16

         #     torch_dtype=torch.float16
         # ).to('cuda')
         model = AutoModelForVision2Seq.from_pretrained(
+            "Aekanun/Llama-3.2-11B-Vision-Instruct-XRay-v1.0",
             load_in_4bit = True,
             device_map="auto",
             torch_dtype = torch.float16