It seems that we can parameterize the width and the height; let's try on dragnuwa (#36)

- It seems that we can parameterize the width and the height; let's try on dragnuwa (24423e8fe404bc27687269c56e6fd9ed6b5e58e6)


Co-authored-by: Fabrice TIERCELIN <[email protected]>

app.py

@@ -109,6 +109,7 @@ def animate(
         gr.update(visible = True)
     ]
 
+@torch.no_grad()
 @spaces.GPU(duration=120)
 def animate_on_gpu(
     image: Image,
@@ -122,7 +123,7 @@ def animate_on_gpu(
     generator = torch.manual_seed(seed)
 
     if version == "dragnuwa":
-        return dragnuwaPipe(image, decode_chunk_size=decoding_t, generator=generator, motion_bucket_id=motion_bucket_id, noise_aug_strength=noise_aug_strength, num_frames=25).frames[0]
+        return dragnuwaPipe(image, width=1024, height=576, decode_chunk_size=decoding_t, generator=generator, motion_bucket_id=motion_bucket_id, noise_aug_strength=noise_aug_strength, num_frames=25).frames[0]
     elif version == "svdxt":
         return fps25Pipe(image, decode_chunk_size=decoding_t, generator=generator, motion_bucket_id=motion_bucket_id, noise_aug_strength=noise_aug_strength, num_frames=25).frames[0]
     else: