Yuanshi commited on
Commit
5ba8bf4
·
verified ·
1 Parent(s): e5bea58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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", subfolder="transformer", torch_dtype=torch.bfloat16
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