Spaces:

ninjals
/

FoodExtract-Vision-v1

Running on Zero

ninjals commited on 29 days ago

Commit

77f345f

verified ·

1 Parent(s): a665c45

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ print(f"[INFO] Loading Original Model")
 original_pipeline = pipeline(
     "image-text-to-text",
     model=BASE_MODEL_ID,
-    dtype=DTYPE,
     device_map="auto",
 )
@@ -25,7 +25,7 @@ print(f"[INFO] Loading Fine-tuned Model")
 ft_pipe = pipeline(
     "image-text-to-text",
     model=FINE_TUNED_MODEL_ID,
-    dtype=DTYPE,
     device_map="auto",
 )

 original_pipeline = pipeline(
     "image-text-to-text",
     model=BASE_MODEL_ID,
+    torch_dtype=DTYPE,
     device_map="auto",
 )
 ft_pipe = pipeline(
     "image-text-to-text",
     model=FINE_TUNED_MODEL_ID,
+    torch_dtype=DTYPE,
     device_map="auto",
 )