Klayand commited on
Commit
1aa304d
·
1 Parent(s): c0f1810

fix error cannot access sd3.5

Browse files
Files changed (2) hide show
  1. app.py +3 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -17,7 +17,9 @@ from diffusion_pipeline.sd35_pipeline import StableDiffusion3Pipeline, FlowMatch
17
  from diffusion_pipeline.sdxl_pipeline import StableDiffusionXLPipeline
18
  from diffusers import BitsAndBytesConfig, SD3Transformer2DModel
19
  from diffusers import FlowMatchEulerDiscreteScheduler, DDIMInverseScheduler, DDIMScheduler
20
-
 
 
21
  device = torch.device('cuda')
22
 
23
 
 
17
  from diffusion_pipeline.sdxl_pipeline import StableDiffusionXLPipeline
18
  from diffusers import BitsAndBytesConfig, SD3Transformer2DModel
19
  from diffusers import FlowMatchEulerDiscreteScheduler, DDIMInverseScheduler, DDIMScheduler
20
+ from huggingface_hub import login
21
+ import os
22
+ login(token=os.getenv('HF_TOKEN'))
23
  device = torch.device('cuda')
24
 
25
 
requirements.txt CHANGED
@@ -6,6 +6,7 @@ accelerate
6
  pandas
7
  imageio
8
  gradio
 
9
  bitsandbytes
10
  imageio-ffmpeg
11
  omegaconf
 
6
  pandas
7
  imageio
8
  gradio
9
+ sentencepiece
10
  bitsandbytes
11
  imageio-ffmpeg
12
  omegaconf