gracekim0513 commited on
Commit
ea92a0b
·
verified ·
1 Parent(s): 8a44205

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
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