:tada: add additional negative embeddings

app.py

@@ -80,7 +80,7 @@ def inference(
         num_inference_steps: int = 50,
         num_images: int = 4,
         safety_checker: bool = True,
-        use_model_offload:bool = False,
+        use_model_offload: bool = False,
         seed: int = 8888,
         progress=gr.Progress(track_tqdm=True),
 ) -> Image.Image:
@@ -105,9 +105,9 @@ def inference(
     if model_id not in DIFFUSERS_MODEL_IDS:
         progress(0.3, 'Loading Textual Inversion...')
         # Load Textual Inversion
-        pipe.load_textual_inversion(
-            "checkpoints/embeddings/BadNegAnatomyV1 neg.pt", token='BadNegAnatomyV1-neg'
-        )
+        pipe.load_textual_inversion('checkpoints/embeddings/BadNegAnatomyV1 neg.pt', token='BadNegAnatomyV1-neg')
+        pipe.load_textual_inversion('checkpoints/embeddings/Deep Negative V1 75T.pt', token='DeepNegativeV1')
+        pipe.load_textual_inversion('checkpoints/embeddings/easynegative.safetensors', token='EasyNegative')
 
     # Generation
     progress(0.4, 'Generating images...')
@@ -161,7 +161,7 @@ if __name__ == "__main__":
                         width = gr.Number(label="Width", value=512, step=64, minimum=64, maximum=2048)
                         height = gr.Number(label="Height", value=512, step=64, minimum=64, maximum=2048)
                         num_images = gr.Number(label="Num Images", value=4, minimum=1, maximum=10, step=1)
-                        seed = gr.Number(label="Seed", value=8888,  step=1)
+                        seed = gr.Number(label="Seed", value=8888, step=1)
 
                     guidance_scale = gr.Slider(label="Guidance Scale", value=7.5, step=0.5, minimum=0, maximum=10)
                     num_inference_step = gr.Slider(
@@ -172,6 +172,20 @@ if __name__ == "__main__":
                         use_safety_checker = gr.Checkbox(value=True, label='Use Safety Checker')
                         use_model_offload = gr.Checkbox(value=False, label='Use Model Offload')
 
+                with gr.Accordion(label='Notes', open=False):
+                    # language=HTML
+                    notes = gr.HTML(
+                        """
+                            <h2>Negative Embeddings</h2>
+                            <p>If you want to use negative embedding, use the following tokens in the prompt.</p>
+                            <ul>
+                                <li><a href='https://civitai.com/models/59614/badneganatomy-textual-inversion'>BadNegAnatomyV1-neg</a></li>
+                                <li><a href='https://civitai.com/models/4629/deep-negative-v1x'>DeepNegativeV1</a> </li>
+                                <li><a href='https://civitai.com/models/7808/easynegative'>EasyNegative</a></li>
+                            </ul>
+                        """
+                    )
+
             with gr.Column():
                 output_image = gr.Image(label="Image", type="pil")
 

checkpoints/embeddings/Deep Negative V1 75T.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:54e7e4826d53949a3d0dde40aea023b1e456a618c608a7630e3999fd38f93245
+size 231339

checkpoints/embeddings/easynegative.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c74b4e810b030f6b75fde959e2db678c268d07115b85356d3c0138ba5eb42340
+size 24655