파라미터 추가

.gitignore

@@ -0,0 +1 @@
+.venv

app.py

@@ -8,7 +8,7 @@ import requests
 import io
 import os
 from PIL import Image
-import spaces
+# import spaces
 
 from diffusers import (
     StableDiffusionPipeline,
@@ -29,14 +29,14 @@ qrcode_generator = qrcode.QRCode(
 )
 
 controlnet = ControlNetModel.from_pretrained(
-    "DionTimmer/controlnet_qrcode-control_v1p_sd15", torch_dtype=torch.float16
+    "DionTimmer/controlnet_qrcode-control_v1p_sd15", torch_dtype=torch.float32
 ) #.to("cuda")
 
 pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
     controlnet=controlnet,
     safety_checker=None,
-    torch_dtype=torch.float16,
+    torch_dtype=torch.float32,
 ) #.to("cuda")
 # pipe.enable_xformers_memory_efficient_attention()
 
@@ -62,7 +62,7 @@ SAMPLER_MAP = {
     "DEIS": lambda config: DEISMultistepScheduler.from_config(config),
 }
 
-@spaces.GPU()
+# @spaces.GPU()
 def inference(
     qr_code_content: str,
     prompt: str,
@@ -72,8 +72,9 @@ def inference(
     strength: float = 0.8,
     seed: int = -1,
     init_image: Image.Image | None = None,
-    custom_image: Image.Image | None = None,
-    use_qr_code_as_init_image = True,
+    control_guidance_start: float = 0,
+    control_guidance_end: float = 1,
+    num_inference_steps: int = 20,
     sampler = "DPM++ Karras SDE",
 ):
     if prompt is None or prompt == "":
@@ -99,9 +100,9 @@ def inference(
 
     qrcode_image = qr.make_image(fill_color="black", back_color="white")
     qrcode_image = resize_for_condition_image(qrcode_image, 768)
-    if custom_image is None:
+    if init_image is None:
         print("Using QR Code Image")
-        custom_image = qrcode_image
+        init_image = qrcode_image
 
     out = pipe(
         prompt=prompt,
@@ -114,7 +115,9 @@ def inference(
         controlnet_conditioning_scale=float(controlnet_conditioning_scale),  # type: ignore
         generator=generator,
         strength=float(strength),
-        num_inference_steps=40,
+        num_inference_steps=num_inference_steps,
+        control_guidance_start=control_guidance_start,
+        control_guidance_end=control_guidance_end,
     )
     return out.images[0]  # type: ignore
 
@@ -143,11 +146,7 @@ model: https://huggingface.co/DionTimmer/controlnet_qrcode-control_v1p_sd15
                 info="QR Code Content or URL",
                 value="",
             )
-            with gr.Accordion(label="init Image (Optional)"):#, open=False):
-                custom_image = gr.Image(
-                    label="Init Code Image  Leave blank to automatically generate QR code",
-                    type="pil",
-                )
+           
 
             prompt = gr.Textbox(
                 label="Prompt",
@@ -157,9 +156,8 @@ model: https://huggingface.co/DionTimmer/controlnet_qrcode-control_v1p_sd15
                 label="Negative Prompt",
                 value="ugly, disfigured, low quality, blurry, nsfw",
             )
-            use_qr_code_as_init_image = gr.Checkbox(label="Use QR code as init image", value=True, interactive=False, info="Whether init image should be QR code. Unclick to pass init image or generate init image with Stable Diffusion 2.1")
 
-            with gr.Accordion(label="Init Images (Optional)", open=False, visible=False) as init_image_acc:
+            with gr.Accordion(label="Init Images (Optional)", open=True):
                 init_image = gr.Image(label="Init Image (Optional). Leave blank to generate image with SD 2.1", type="pil")
 
 
@@ -174,12 +172,12 @@ model: https://huggingface.co/DionTimmer/controlnet_qrcode-control_v1p_sd15
                     value=1.1,
                     label="Controlnet Conditioning Scale",
                 )
-                steps = gr.Slider(
+                num_inference_steps = gr.Slider(
                     minimum=20,
                     maximum=50,
                     step=1,
                     value=20,
-                    label="Steps",
+                    label="num_inference_steps",
                 )
                 strength = gr.Slider(
                     minimum=0.0, maximum=1.0, step=0.01, value=0.9, label="Strength"
@@ -229,8 +227,9 @@ model: https://huggingface.co/DionTimmer/controlnet_qrcode-control_v1p_sd15
             strength,
             seed,
             init_image,
-            custom_image,
-            use_qr_code_as_init_image,
+            control_guidance_start,
+            control_guidance_end,
+            num_inference_steps,
             sampler,
         ],
         outputs=[result_image],
@@ -248,36 +247,37 @@ model: https://huggingface.co/DionTimmer/controlnet_qrcode-control_v1p_sd15
                 0.9,
                 5392011833,
                 None,
-                None,
-                True,
-                "DPM++ Karras SDE",
-            ],
-            [
-                "https://huggingface.co/",
-                "Bright sunshine coming through the cracks of a wet, cave wall of big rocks",
-                "ugly, disfigured, low quality, blurry, nsfw",
-                7.5,
-                1.11,
-                0.9,
-                2523992465,
-                None,
-                None,
-                True,
-                "DPM++ Karras SDE",
-            ],
-            [
-                "https://huggingface.co/",
-                "Sky view of highly aesthetic, ancient greek thermal baths  in beautiful nature",
-                "ugly, disfigured, low quality, blurry, nsfw",
-                7.5,
-                1.5,
-                0.9,
-                2523992465,
-                None,
-                None,
-                True,
+                0,
+                1,
+                20,
                 "DPM++ Karras SDE",
             ],
+            # [
+            #     "https://huggingface.co/",
+            #     "Bright sunshine coming through the cracks of a wet, cave wall of big rocks",
+            #     "ugly, disfigured, low quality, blurry, nsfw",
+            #     7.5,
+            #     1.11,
+            #     0.9,
+            #     2523992465,
+            #     None,
+            #     None,
+            #     True,
+            #     "DPM++ Karras SDE",
+            # ],
+            # [
+            #     "https://huggingface.co/",
+            #     "Sky view of highly aesthetic, ancient greek thermal baths  in beautiful nature",
+            #     "ugly, disfigured, low quality, blurry, nsfw",
+            #     7.5,
+            #     1.5,
+            #     0.9,
+            #     2523992465,
+            #     None,
+            #     None,
+            #     True,
+            #     "DPM++ Karras SDE",
+            # ],
         ],
         fn=inference,
         inputs=[
@@ -289,8 +289,9 @@ model: https://huggingface.co/DionTimmer/controlnet_qrcode-control_v1p_sd15
             strength,
             seed,
             init_image,
-            custom_image,
-            use_qr_code_as_init_image,
+            control_guidance_start,
+            control_guidance_end,
+            num_inference_steps,
             sampler,
         ],
         outputs=[result_image],

requirements.txt

@@ -1,9 +1,7 @@
-diffusers
-transformers
+diffusers==0.25.0
+transformers==4.27.2
 accelerate
-torch
-xformers
-gradio
+torch==2.0.1
 Pillow
 qrcode
 gradio==4.8.0