Spaces:

Prgckwb
/

stable-diffusion-demo

Sleeping

Prgckwb commited on Jun 17

Commit

750ec07

•

1 Parent(s): 5244e66

:tada: add negative embedding

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,7 @@ if device == 'cuda':
         current_model_id,
         torch_dtype=dtype,
     )
-    pipe.enable_sequential_cpu_offload()
 @dataclasses.dataclass
@@ -102,7 +102,7 @@ def inference(
             model_id,
             torch_dtype=dtype,
         )
-        pipe.enable_sequential_cpu_offload()
         current_model_id = model_id

         current_model_id,
         torch_dtype=dtype,
     )
+    pipe.enable_model_cpu_offload()
 @dataclasses.dataclass
             model_id,
             torch_dtype=dtype,
         )
+        pipe.enable_model_cpu_offload()
         current_model_id = model_id