Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev
|
|
16 |
|
17 |
# New Here
|
18 |
turbo = FluxTransformer2DModel.from_pretrained(
|
19 |
-
"Yuanshi/FLUX.1-Kontext-Turbo",
|
20 |
).to("cuda")
|
21 |
pipe.transformer = turbo
|
22 |
|
|
|
16 |
|
17 |
# New Here
|
18 |
turbo = FluxTransformer2DModel.from_pretrained(
|
19 |
+
"Yuanshi/FLUX.1-Kontext-Turbo", torch_dtype=torch.bfloat16
|
20 |
).to("cuda")
|
21 |
pipe.transformer = turbo
|
22 |
|