Spaces:

Sijuade
/

Stable-Diffusion

Sleeping

Sijuade commited on Dec 17, 2023

Commit

ac47027

•

1 Parent(s): 7027d50

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -21,7 +21,7 @@ elastic_transformer = T.ElasticTransform(alpha=550.0, sigma=5.0)
 pretrained_model_name_or_path = "segmind/tiny-sd"
 pipe = DiffusionPipeline.from_pretrained(
     pretrained_model_name_or_path,
-    low_cpu_mem_usage = True
 ).to(torch_device)

 pretrained_model_name_or_path = "segmind/tiny-sd"
 pipe = DiffusionPipeline.from_pretrained(
     pretrained_model_name_or_path,
+    #low_cpu_mem_usage = True
 ).to(torch_device)