Update app.py

app.py

@@ -114,25 +114,25 @@ snapshot_download(
 
 
 snapshot_download(
-    repo_id="stabilityai/stable-diffusion-2-1-base", 
-    local_dir="preset/models/stable-diffusion-2-1-base"
+    repo_id="stabilityai/sd-turbo", 
+    local_dir="preset/models/sd-turbo"
 )
 
+
 snapshot_download(
     repo_id="xinyu1205/recognize_anything_model", 
     local_dir="preset/models/"
 )
 
-
 # Load scheduler, tokenizer and models.
-pretrained_model_path = 'preset/models/stable-diffusion-2-1-base'
+pretrained_model_path = 'preset/models/sd-turbo'
 seesr_model_path = 'preset/models/seesr'
 
 scheduler = DDIMScheduler.from_pretrained(pretrained_model_path, subfolder="scheduler")
 text_encoder = CLIPTextModel.from_pretrained(pretrained_model_path, subfolder="text_encoder")
 tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
 vae = AutoencoderKL.from_pretrained(pretrained_model_path, subfolder="vae")
-feature_extractor = CLIPImageProcessor.from_pretrained(f"{pretrained_model_path}/feature_extractor")
+# feature_extractor = CLIPImageProcessor.from_pretrained(f"{pretrained_model_path}/feature_extractor")
 unet = UNet2DConditionModel.from_pretrained(seesr_model_path, subfolder="unet")
 controlnet = ControlNetModel.from_pretrained(seesr_model_path, subfolder="controlnet")
 
@@ -192,9 +192,9 @@ def magnify(
     user_prompt = "",
     positive_prompt = "clean, high-resolution, 8k, best quality, masterpiece",
     negative_prompt = "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
-    num_inference_steps = 50,
+    num_inference_steps = 2,
     scale_factor = 4,
-    cfg_scale = 7.5,
+    cfg_scale = 1,
     seed = 123,
     latent_tiled_size = 320,
     latent_tiled_overlap = 4,
@@ -302,8 +302,8 @@ with gr.Blocks(css=css, theme=theme) as demo:
                         label="Negative Prompt",
                         value="dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality"
                     )
-                    cfg_scale = gr.Slider(label="Classifier Free Guidance Scale (Set to 1.0 in sd-turbo)", minimum=1, maximum=10, value=7.5, step=0)
-                    num_inference_steps = gr.Slider(label="Inference Steps", minimum=2, maximum=100, value=50, step=1)
+                    cfg_scale = gr.Slider(label="Classifier Free Guidance Scale (Set to 1.0 in sd-turbo)", minimum=1, maximum=10, value=1, step=0)
+                    num_inference_steps = gr.Slider(label="Inference Steps", minimum=2, maximum=100, value=2, step=1)
                     seed = gr.Slider(label="Seed", minimum=-1, maximum=2147483647, step=1, value=231)
                     sample_times = gr.Slider(label="Sample Times", minimum=1, maximum=10, step=1, value=1)
                     latent_tiled_size = gr.Slider(label="Diffusion Tile Size", minimum=128, maximum=480, value=320, step=1)