:tada: change some process

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import spaces
 import torch
 from PIL import Image
+from compel import Compel, DiffusersTextualInversionManager
 from diffusers import DiffusionPipeline
 from diffusers.utils import make_image_grid
 
@@ -51,6 +52,9 @@ def load_pipeline(model_id, use_model_offload, safety_checker):
     else:
         pipe = pipe.to(device)
 
+    if not safety_checker:
+        pipe.safety_checker = None
+
     return pipe
 
 
@@ -73,12 +77,22 @@ def inference(
     progress(0, 'Loading pipeline...')
     pipe = load_pipeline(model_id, use_model_offload, safety_checker)
 
+    # For Compel
+    textual_inversion_manager = DiffusersTextualInversionManager(pipe)
+    compel_procs = Compel(
+        tokenizer=pipe.tokenizer,
+        text_encoder=pipe.text_encoder,
+        textual_inversion_manager=textual_inversion_manager,
+    )
+    prompt_embed = compel_procs(prompt)
+    negative_prompt_embed = compel_procs(negative_prompt)
+
     generator = torch.Generator(device=device).manual_seed(seed)
 
     progress(0.3, 'Generating images...')
     images = pipe(
-        prompt,
-        negative_prompt=negative_prompt,
+        prompt_embeds=prompt_embed,
+        negative_prompt_embeds=negative_prompt_embed,
         width=width,
         height=height,
         guidance_scale=guidance_scale,
@@ -164,7 +178,7 @@ def build_interface():
             seed,
         ]
 
-        btn = gr.Button("Generate")
+        btn = gr.Button("Generate", variant='primary')
         btn.click(
             fn=inference,
             inputs=inputs,

checkpoints/lora/Hand v3 SD1.5.safetensors

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

checkpoints/lora/detailed style SD1.5.safetensors

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

checkpoints/lora/perfection style SD1.5.safetensors

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

src/example.py

@@ -37,9 +37,9 @@ EXAMPLES = [
     Example(
         prompt='Cinematic Photo of a beautiful korean fashion model bokeh train',
         model_id='Beautiful Realistic Asians',
-        negative_prompt='worst_quality, BadNegAnatomyV1-neg, bradhands cartoon, cgi, render, illustration, painting, drawing',
-        width=512,
-        height=512,
+        negative_prompt='(worst_quality)++, (BadNegAnatomyV1-neg), bradhands cartoon, cgi, render, illustration, painting, drawing',
+        width=768,
+        height=960,
         guidance_scale=5.0,
         num_inference_step=50,
     ).to_list()