Spaces:
Build error
Build error
fix error cannot access sd3.5
Browse files- app.py +3 -1
- 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
|