Upload 2 files

florence2_sd3_tagger8.py

@@ -46,7 +46,7 @@ def fl_run_example(image, fl_model, fl_processor):
     if image.mode != "RGB":
         image = image.convert("RGB")
 
-    inputs = fl_processor(text=prompt, images=image, return_tensors="pt").to("cuda")
+    inputs = fl_processor(text=prompt, images=image, return_tensors="pt")
     generated_ids = fl_model.generate(
         input_ids=inputs["input_ids"],
         pixel_values=inputs["pixel_values"],