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
|