:tada: add negative embedding

README.md

@@ -1,12 +0,0 @@
----
-title: Stable Diffusion Demo
-emoji: 🌖
-colorFrom: pink
-colorTo: yellow
-sdk: gradio
-sdk_version: 4.36.1
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -33,8 +33,7 @@ if device == 'cuda':
         current_model_id,
         torch_dtype=dtype,
     ).to(device)
-else:
-    dtype = torch.float32
+
 
 
 @dataclasses.dataclass
@@ -103,7 +102,8 @@ def inference(
             pipe = DiffusionPipeline.from_pretrained(
                 model_id,
                 torch_dtype=dtype,
-            ).to(device)
+            )
+            pipe.enable_model_cpu_offload()
 
             current_model_id = model_id
         except Exception as e:

requirements.txt

@@ -9,3 +9,4 @@ accelerate
 sentencepiece
 spaces
 Pillow
+protobuf