Remove torch_dtype

app.py

@@ -30,7 +30,6 @@ else:
     state_dict, network_alphas = FluxFillPipeline.lora_state_dict(
         pretrained_model_name_or_path_or_dict="blanchon/FluxFillFurniture",
         weight_name="pytorch_lora_weights3.safetensors",
-        torch_dtype=torch.bfloat16,
         return_alphas=True,
     )
 
@@ -38,13 +37,12 @@ else:
         msg = "Invalid LoRA checkpoint."
         raise ValueError(msg)
 
-    pipe = FluxFillPipeline.from_pretrained(
-        "black-forest-labs/FLUX.1-Fill-dev", torch_dtype=torch.bfloat16
-    ).to(DEVICE)
+    pipe = FluxFillPipeline.from_pretrained("black-forest-labs/FLUX.1-Fill-dev").to(
+        DEVICE
+    )
     FluxFillPipeline.load_lora_into_transformer(
         state_dict=state_dict,
         network_alphas=network_alphas,
-        torch_dtype=torch.bfloat16,
         transformer=pipe.transformer,
     )
     pipe.to(DEVICE)