Fix cuda

README.md

@@ -9,7 +9,7 @@ pinned: false
 license: openrail
 hf_oauth: true
 disable_embedding: true
-short_description: Generate stunning high quality illusion artwork
+short_description: Generate high quality illusion artwork from a pattern and a prompt
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -35,10 +35,15 @@ controlnet = ControlNetModel.from_pretrained("monster-labs/control_v1p_sd15_qrco
 SAFETY_CHECKER_ENABLED = os.environ.get("SAFETY_CHECKER", "0") == "1"
 safety_checker = None
 feature_extractor = None
+device='cuda'
+device='cpu'
+
+
 if SAFETY_CHECKER_ENABLED:
-    safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker").to("cuda")
+    safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker").to(device)
     feature_extractor = CLIPImageProcessor.from_pretrained("openai/clip-vit-base-patch32")
 
+
 main_pipe = StableDiffusionControlNetPipeline.from_pretrained(
     BASE_MODEL,
     controlnet=controlnet,
@@ -46,7 +51,7 @@ main_pipe = StableDiffusionControlNetPipeline.from_pretrained(
     safety_checker=safety_checker,
     feature_extractor=feature_extractor,
     torch_dtype=torch.float16,
-).to("cuda")
+).to(device)
 
 # Function to check NSFW images
 #def check_nsfw_images(images: list[Image.Image]) -> tuple[list[Image.Image], list[bool]]:
@@ -164,7 +169,7 @@ def inference(
 
     main_pipe.scheduler = SAMPLER_MAP[sampler](main_pipe.scheduler.config)
     my_seed = random.randint(0, 2**32 - 1) if seed == -1 else seed
-    generator = torch.Generator(device="cuda").manual_seed(my_seed)
+    generator = torch.Generator(device=device).manual_seed(my_seed)
     
     out = main_pipe(
         prompt=prompt,
@@ -221,11 +226,15 @@ with gr.Blocks() as app:
     gr.Markdown(
         '''
         <div style="text-align: center;">
-            <h1>Illusion Diffusion HQ 🌀</h1>
+            <h1>pattern + prompt = image</h1>
             <p style="font-size:16px;">Generate stunning high quality illusion artwork with Stable Diffusion</p>
             <p>Illusion Diffusion is back up with a safety checker! Because I have been asked, if you would like to support me, consider using <a href="https://deforum.studio">deforum.studio</a></p>
-            <p>A space by AP <a href="https://twitter.com/angrypenguinPNG">Follow me on Twitter</a> with big contributions from <a href="https://twitter.com/multimodalart">multimodalart</a></p>
-            <p>This project works by using <a href="https://huggingface.co/monster-labs/control_v1p_sd15_qrcode_monster">Monster Labs QR Control Net</a>. Given a prompt and your pattern, we use a QR code conditioned controlnet to create a stunning illusion! Credit to: <a href="https://twitter.com/MrUgleh">MrUgleh</a> for discovering the workflow :)</p>
+            <p>With big contributions from 
+            <ul>
+            <li><a href="https://twitter.com/multimodalart">multimodalart</a></li>
+            <li><a href="https://huggingface.co/monster-labs/control_v1p_sd15_qrcode_monster">Monster Labs QR Control Net</a></li>
+            <li><a href="https://twitter.com/MrUgleh">MrUgleh</a>/li>
+            </ul>
         </div>
         '''
     )
@@ -235,7 +244,7 @@ with gr.Blocks() as app:
     state_img_output = gr.State()
     with gr.Row():
         with gr.Column():
-            control_image = gr.Image(label="Input Illusion", type="pil", elem_id="control_image")
+            control_image = gr.Image(label="Input pattern", type="pil", elem_id="control_image")
             controlnet_conditioning_scale = gr.Slider(minimum=0.0, maximum=5.0, step=0.01, value=0.8, label="Illusion strength", elem_id="illusion_strength", info="ControlNet conditioning scale")
             gr.Examples(examples=["checkers.png", "checkers_mid.jpg", "pattern.png", "ultra_checkers.png", "spiral.jpeg", "funky.jpeg" ], inputs=control_image)
             prompt = gr.Textbox(label="Prompt", elem_id="prompt", info="Type what you want to generate", placeholder="Medieval village scene with busy streets and castle in the distance")
@@ -250,7 +259,7 @@ with gr.Blocks() as app:
                 used_seed = gr.Number(label="Last seed used",interactive=False)
             run_btn = gr.Button("Run")
         with gr.Column():
-            result_image = gr.Image(label="Illusion Diffusion Output", interactive=False, elem_id="output")
+            result_image = gr.Image(label="Output", interactive=False, elem_id="output")
             with gr.Group(elem_id="share-btn-container", visible=False) as share_group:
                 community_icon = gr.HTML(community_icon_html)
                 loading_icon = gr.HTML(loading_icon_html)