Update app.py

app.py

@@ -128,14 +128,11 @@ def inference(source_images,
         text_prompt, negative_prompt, 
         pipe, vae, guidance_scale, 
         h, w, random_seed)->List[PIL.Image.Image]:
-    print("来到inference里")
     torch.manual_seed(random_seed)
-    print("设置完随机种子，准备移动pipe各组件到gpu")
     
     pipe.vae.to(DEVICE)
     pipe.transformer.to(DEVICE)
     pipe.controlnet_transformer.to(DEVICE)
-    print("pipe.vae/transformer/controlnet移动到GPU里")
 
     source_pixel_values = source_images/127.5 - 1.0
     source_pixel_values = source_pixel_values.to(torch.float16).to(DEVICE)
@@ -164,7 +161,6 @@ def inference(source_images,
             image_latents = None
             latents = source_latents
 
-    print("准备调用pipe()函数")
     video = pipe(
         prompt = text_prompt,
         negative_prompt = negative_prompt,
@@ -173,7 +169,7 @@ def inference(source_images,
         height = h,
         width = w,
         num_frames = f,
-        num_inference_steps = 50,#!!!
+        num_inference_steps = 30,
         interval = 6,
         guidance_scale = guidance_scale,
         generator = torch.Generator(device=DEVICE).manual_seed(random_seed)
@@ -220,7 +216,6 @@ def process_video(video_file, image_file, positive_prompt, negative_prompt, guid
             source_images=source_images
             target_images=first_frame[None,None,...]
             
-        print("所有事都干完了，准备inference。")
         video+=inference(source_images, \
             target_images, positive_prompt, \
             negative_prompt, pipe, pipe.vae, \