LPX55 commited on
Commit
c692ff8
·
verified ·
1 Parent(s): 825bcd3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -125,8 +125,9 @@ def resize_image_to_bucket(image: Union[Image.Image, np.ndarray], bucket_reso: T
125
  image = image[crop_top:crop_top + bucket_height, crop_left:crop_left + bucket_width]
126
  return image
127
 
128
- # @spaces.GPU(duration=120)
129
- @torch.inference_mode()
 
130
  def generate_video(prompt: str, frame1: Image.Image, frame2: Image.Image, resolution: str, guidance_scale: float, num_frames: int, num_inference_steps: int) -> bytes:
131
  # Debugging print statements
132
  print(f"Frame 1 Type: {type(frame1)}")
 
125
  image = image[crop_top:crop_top + bucket_height, crop_left:crop_left + bucket_width]
126
  return image
127
 
128
+ #
129
+ # @torch.inference_mode()
130
+ @spaces.GPU(duration=120)
131
  def generate_video(prompt: str, frame1: Image.Image, frame2: Image.Image, resolution: str, guidance_scale: float, num_frames: int, num_inference_steps: int) -> bytes:
132
  # Debugging print statements
133
  print(f"Frame 1 Type: {type(frame1)}")