Manjushri commited on
Commit
437b114
1 Parent(s): e3f976b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,7 +32,7 @@ def sample(
32
  randomize_seed: bool = True,
33
  motion_bucket_id: int = 127,
34
  fps_id: int = 6,
35
- version: str = "svd_xt",
36
  cond_aug: float = 0.02,
37
  decoding_t: int = 3, # Number of frames decoded at a time! This eats most VRAM. Reduce if necessary.
38
  device: str = "cuda",
@@ -55,7 +55,7 @@ def sample(
55
  torch.cuda.empty_cache()
56
  return video_path, seed
57
 
58
- def resize_image(image, output_size=(1024, 578)):
59
  # Calculate aspect ratios
60
  target_aspect = output_size[0] / output_size[1] # Aspect ratio of the desired size
61
  image_aspect = image.width / image.height # Aspect ratio of the original image
 
32
  randomize_seed: bool = True,
33
  motion_bucket_id: int = 127,
34
  fps_id: int = 6,
35
+ version: str = "svd_xt_1-1",
36
  cond_aug: float = 0.02,
37
  decoding_t: int = 3, # Number of frames decoded at a time! This eats most VRAM. Reduce if necessary.
38
  device: str = "cuda",
 
55
  torch.cuda.empty_cache()
56
  return video_path, seed
57
 
58
+ def resize_image(image, output_size=(768, 512)):
59
  # Calculate aspect ratios
60
  target_aspect = output_size[0] / output_size[1] # Aspect ratio of the desired size
61
  image_aspect = image.width / image.height # Aspect ratio of the original image