yiren98 commited on
Commit
3be0f17
·
1 Parent(s): 0fc1f2c
Files changed (1) hide show
  1. gradio_app.py +4 -0
gradio_app.py CHANGED
@@ -27,6 +27,10 @@ CLIP_L_PATH="/tiamat-NAS/hailong/storage_backup/models/stabilityai/stable-diffus
27
  T5XXL_PATH="/tiamat-NAS/hailong/storage_backup/models/stabilityai/stable-diffusion-3-medium/text_encoders/t5xxl_fp16.safetensors"
28
  AE_PATH="/tiamat-vePFS/share_data/storage/huggingface/models/black-forest-labs/FLUX.1-dev/ae.safetensors"
29
 
 
 
 
 
30
  from huggingface_hub import hf_hub_download
31
  repo_id = "Kijai/flux-fp8"
32
  file_name = "flux1-dev-fp8.safetensors"
 
27
  T5XXL_PATH="/tiamat-NAS/hailong/storage_backup/models/stabilityai/stable-diffusion-3-medium/text_encoders/t5xxl_fp16.safetensors"
28
  AE_PATH="/tiamat-vePFS/share_data/storage/huggingface/models/black-forest-labs/FLUX.1-dev/ae.safetensors"
29
 
30
+ from huggingface_hub import login
31
+ hf_token = os.getenv("HF_TOKEN")
32
+ login(token=hf_token)
33
+
34
  from huggingface_hub import hf_hub_download
35
  repo_id = "Kijai/flux-fp8"
36
  file_name = "flux1-dev-fp8.safetensors"