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 |
|