Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -23,6 +23,13 @@ import sampling
|
|
| 23 |
from modules.autoencoder import AutoEncoder
|
| 24 |
from modules.conditioner import Qwen25VL_7b_Embedder as Qwen2VLEmbedder
|
| 25 |
from modules.model_edit import Step1XParams, Step1XEdit
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
|
| 27 |
print("TORCH_CUDA", torch.cuda.is_available())
|
| 28 |
|
|
|
|
| 23 |
from modules.autoencoder import AutoEncoder
|
| 24 |
from modules.conditioner import Qwen25VL_7b_Embedder as Qwen2VLEmbedder
|
| 25 |
from modules.model_edit import Step1XParams, Step1XEdit
|
| 26 |
+
from diffusers import StableDiffusionPipeline
|
| 27 |
+
|
| 28 |
+
pipe = StableDiffusionPipeline.from_pretrained(
|
| 29 |
+
"stabilityai/stable-diffusion-2",
|
| 30 |
+
cache_dir="/tmp", # 작게 제한된 디렉토리 사용
|
| 31 |
+
local_files_only=False
|
| 32 |
+
)
|
| 33 |
|
| 34 |
print("TORCH_CUDA", torch.cuda.is_available())
|
| 35 |
|