Manjushri commited on
Commit
cfd486f
·
verified ·
1 Parent(s): b14bfaf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,7 +29,7 @@ def sample(
29
  version: str = "svd_xt",
30
  cond_aug: float = 0.02,
31
  decoding_t: int = 3, # Number of frames decoded at a time! This eats most VRAM. Reduce if necessary.
32
- device: str = "cpu",
33
  output_folder: str = "outputs",
34
  ):
35
  if image.mode == "RGBA":
@@ -49,7 +49,7 @@ def sample(
49
 
50
  return video_path, seed
51
 
52
- def resize_image(image, output_size=(512, 256)):
53
  # Calculate aspect ratios
54
  target_aspect = output_size[0] / output_size[1] # Aspect ratio of the desired size
55
  image_aspect = image.width / image.height # Aspect ratio of the original image
 
29
  version: str = "svd_xt",
30
  cond_aug: float = 0.02,
31
  decoding_t: int = 3, # Number of frames decoded at a time! This eats most VRAM. Reduce if necessary.
32
+ device: str = "cuda",
33
  output_folder: str = "outputs",
34
  ):
35
  if image.mode == "RGBA":
 
49
 
50
  return video_path, seed
51
 
52
+ def resize_image(image, output_size=(1024, 578)):
53
  # Calculate aspect ratios
54
  target_aspect = output_size[0] / output_size[1] # Aspect ratio of the desired size
55
  image_aspect = image.width / image.height # Aspect ratio of the original image