Update app.py

app.py

@@ -128,8 +128,10 @@ 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)
@@ -209,7 +211,6 @@ def process_video(video_file, image_file, positive_prompt, negative_prompt, guid
     
     video:List[PIL.Image.Image]=[]
 
-    print("所有事都干完了，准备inference。")
     for i in progress.tqdm(range(video_shard)):
         if i>0: #first frame guidence
             first_frame=transforms.ToTensor()(video[-1])
@@ -217,7 +218,8 @@ def process_video(video_file, image_file, positive_prompt, negative_prompt, guid
             first_frame = rearrange(first_frame,"c w h -> w h c")
             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, \