Axel-Student commited on
Commit
6fc88c7
·
1 Parent(s): 424839d

login in pipe

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,7 +18,8 @@ image = pipe(
18
  guidance_scale=3.5,
19
  num_inference_steps=50,
20
  max_sequence_length=512,
21
- generator=torch.Generator("cpu").manual_seed(0)
 
22
  ).images[0]
23
  image.save("flux-dev.png")
24
 
 
18
  guidance_scale=3.5,
19
  num_inference_steps=50,
20
  max_sequence_length=512,
21
+ generator=torch.Generator("cpu").manual_seed(0),
22
+ use_auth_token=os.getenv("HF_TOKEN")
23
  ).images[0]
24
  image.save("flux-dev.png")
25