bugfix

app.py

@@ -276,6 +276,7 @@ def process(
                 image_resolution=width,
                 detect_resolution=512,
             )
+            
 
         if control_mode == "scribble_hed":
             preprocessor.load("HED")
@@ -284,7 +285,8 @@ def process(
                 image_resolution=image_resolution,
                 detect_resolution=preprocess_resolution,
             )
-        
+    
+    control_image = control_image.resize((width, height), Image.LANCZOS)    
     control_mode_id = control_mode_ids[control_mode]
 
     try:
@@ -360,7 +362,7 @@ with gr.Blocks() as demo:
                 )
             
             control_mode = gr.Dropdown(
-                [ "canny", "depth_midas",  "openpose", "mlsd", "low_quality", "gray", "blur", "tile"], label="Controlnet Model", info="choose controlnet model!", value="canny"
+                [ "canny", "depth_midas",  "openpose", "mlsd"], label="Controlnet Model", info="choose controlnet model!", value="openpose"
             )
 
             submit_button_component = gr.Button(value='Submit', variant='primary', scale=0)

requirements.txt

@@ -1,6 +1,7 @@
 accelerate
 invisible_watermark
-torch
+torch==2.0.1
+torchvision==0.15.2
 transformers
 xformers
 tqdm
@@ -18,4 +19,6 @@ boto3
 sentencepiece
 peft
 controlnet-aux
-kornia
+kornia
+deepspeed
+einops