FLUX.1-Kontext-Turbo

Running on Zero

Yuanshi commited on Jul 14

Commit

5ba8bf4

verified ·

1 Parent(s): e5bea58

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev
 # New Here
 turbo = FluxTransformer2DModel.from_pretrained(
-    "Yuanshi/FLUX.1-Kontext-Turbo", subfolder="transformer", torch_dtype=torch.bfloat16
 ).to("cuda")
 pipe.transformer = turbo

 # New Here
 turbo = FluxTransformer2DModel.from_pretrained(
+    "Yuanshi/FLUX.1-Kontext-Turbo", torch_dtype=torch.bfloat16
 ).to("cuda")
 pipe.transformer = turbo