Spaces:

Kontext-Style
/

Kontext-Style-LoRAs

Running on Zero

Xsong123 commited on Jul 5

Commit

5d4e1e8

verified ·

1 Parent(s): b0c1f73

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ MAX_SEED = np.iinfo(np.int32).max
 # Use the local path for the base model as in your test.py
 pipe = FluxKontextPipeline.from_pretrained(
-    "/hpc2hdd/home/sfei285/Project/Editing/FLUX.1-Kontext-dev",
     torch_dtype=torch.bfloat16
 ).to("cuda")

 # Use the local path for the base model as in your test.py
 pipe = FluxKontextPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-Kontext-dev",
     torch_dtype=torch.bfloat16
 ).to("cuda")