Spaces:

X-HighVoltage-X
/

Flux.1-Fill-dev-Inpainting-Super-Realism-LoRA

Running on Zero

X-HighVoltage-X commited on Oct 7

Commit

f36cdfd

verified ·

1 Parent(s): bbd5197

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,9 @@ from loras import LoRA, loras
 MAX_SEED = np.iinfo(np.int32).max
 pipe = FluxFillPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-Fill-dev", torch_dtype=torch.bfloat16
 )
 flux_keywords_available = [

 MAX_SEED = np.iinfo(np.int32).max
 pipe = FluxFillPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-Fill-dev",
+    torch_dtype=torch.bfloat16,
+    device_map="auto"
 )
 flux_keywords_available = [