RefurnishAI

Runtime error

Ashoka74 commited on 13 days ago

Commit

3e7a57f

•

1 Parent(s): 2958beb

Update app_merged.py

Files changed (1) hide show

app_merged.py CHANGED Viewed

@@ -194,8 +194,8 @@ transformer = FluxTransformer2DModel.from_single_file(
     torch_dtype=torch.bfloat16,
 )
-fill_pipe = FluxPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-dev",
     transformer=transformer,
     generator=torch.manual_seed(0),
     torch_dtype=torch.bfloat16,

     torch_dtype=torch.bfloat16,
 )
+fill_pipe = FluxFillPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-Fill-dev",
     transformer=transformer,
     generator=torch.manual_seed(0),
     torch_dtype=torch.bfloat16,