Axel-Student commited on
Commit
43c767e
·
1 Parent(s): d35479f

trying to add auth

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
  from diffusers import FluxPipeline # type: ignore
3
  import gradio as gr # type: ignore
4
  from huggingface_hub import login
5
- login(token="HF_TOKEN")
6
 
7
  pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
8
  pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
 
2
  from diffusers import FluxPipeline # type: ignore
3
  import gradio as gr # type: ignore
4
  from huggingface_hub import login
5
+ login()
6
 
7
  pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
8
  pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power