hatmanstack commited on
Commit
93714a0
·
1 Parent(s): 28111ae

HuggingLogin Einops

Browse files
Files changed (2) hide show
  1. app.py +5 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -6,6 +6,11 @@ from PIL import Image
6
  from models_transformer_sd3 import SD3Transformer2DModel
7
  from pipeline_stable_diffusion_3_ipa import StableDiffusion3Pipeline
8
  import gc
 
 
 
 
 
9
 
10
  model_path = 'stabilityai/stable-diffusion-3.5-large'
11
  ip_adapter_path = './ip-adapter.bin'
 
6
  from models_transformer_sd3 import SD3Transformer2DModel
7
  from pipeline_stable_diffusion_3_ipa import StableDiffusion3Pipeline
8
  import gc
9
+ import os
10
+ from huggingface_hub import login
11
+
12
+ TOKEN = os.getenv('TOKEN')
13
+ login(TOKEN)
14
 
15
  model_path = 'stabilityai/stable-diffusion-3.5-large'
16
  ip_adapter_path = './ip-adapter.bin'
requirements.txt CHANGED
@@ -2,4 +2,5 @@ diffusers
2
  torch
3
  transformers
4
  accelerate
5
- Pillow
 
 
2
  torch
3
  transformers
4
  accelerate
5
+ Pillow
6
+ einops