Update modules/masking_module.py

modules/masking_module.py

@@ -77,7 +77,7 @@ def florence2(image,task_prompt, text_input=None):
     else:
         prompt = task_prompt + text_input
 
-    input_florence = processor(text=prompt, images=image, return_tensors="pt").to(torch.float16).to("cuda")
+    input_florence = processor(text=prompt, images=image, return_tensors="pt").to("cuda")
     print(input_florence)
     generated_ids = model.generate(
         input_ids=input_florence["input_ids"],