test

app.py

@@ -71,7 +71,7 @@ class AnimateController:
                 right = original_width
                 bottom = top + new_height
                 pil_image = pil_image.crop((left, top, right, bottom))
-                
+
         pil_image = pil_image.resize((width_slider, height_slider))
         return gr.Textbox(value=img_path), gr.Image(value=np.array(pil_image))
 
@@ -142,6 +142,7 @@ class AnimateController:
         
         first_frame = img_transform(first_frame).unsqueeze(0)
         first_frame = first_frame.to("cuda")
+        print("first_frame", first_frame.device)
 
         if use_frameinit:
             self.pipeline.init_filter(

requirements.txt

@@ -1,6 +1,5 @@
 torch==2.1.0
 torchvision==0.16.0
-torchaudio==2.1.0
 diffusers==0.21.2
 transformers==4.25.1
 accelerate==0.23.0