Yaron Koresh commited on
Commit
353e7c2
·
verified ·
1 Parent(s): 9db21d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ from diffusers.pipelines.flux import FluxPipeline
17
 
18
  device = "cuda" if torch.cuda.is_available() else "cpu"
19
  dtype = torch.float16
20
- pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype).to(device)
21
  pipe.enable_vae_slicing()
22
  pipe.enable_vae_tiling()
23
 
 
17
 
18
  device = "cuda" if torch.cuda.is_available() else "cpu"
19
  dtype = torch.float16
20
+ pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
21
  pipe.enable_vae_slicing()
22
  pipe.enable_vae_tiling()
23