naykun commited on
Commit
59bff29
·
1 Parent(s): 01e40c4
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -14,6 +14,9 @@ import gradio as gr
14
  LOG_DIR = "/tmp/local"
15
  MAX_SEED = np.iinfo(np.int32).max
16
 
 
 
 
17
  pipeline = QwenImageLayeredPipeline.from_pretrained("Qwen/Qwen-Image-Layered")
18
  pipeline = pipeline.to("cuda", torch.bfloat16)
19
  pipeline.set_progress_bar_config(disable=None)
 
14
  LOG_DIR = "/tmp/local"
15
  MAX_SEED = np.iinfo(np.int32).max
16
 
17
+ from huggingface_hub import login
18
+ login(token=os.environ.get('hf'))
19
+
20
  pipeline = QwenImageLayeredPipeline.from_pretrained("Qwen/Qwen-Image-Layered")
21
  pipeline = pipeline.to("cuda", torch.bfloat16)
22
  pipeline.set_progress_bar_config(disable=None)