1inkusFace commited on
Commit
72a5e4c
·
verified ·
1 Parent(s): cdf45cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -113,7 +113,7 @@ def generate(segment, image, prompt, size, guidance_scale, num_inference_steps,
113
  segment_timesteps = torch.from_numpy(timesteps_split_np[0]).to("cuda")
114
  num_channels_latents = pipe.transformer.config.in_channels
115
  num_channels_latents = int(num_channels_latents / 2)
116
- image = Image.open(image.name).convert('RGB')
117
  image.resize((size,size), Image.LANCZOS)
118
  with torch.no_grad():
119
  image = pipe.video_processor.preprocess(image, height=size, width=size).to(
 
113
  segment_timesteps = torch.from_numpy(timesteps_split_np[0]).to("cuda")
114
  num_channels_latents = pipe.transformer.config.in_channels
115
  num_channels_latents = int(num_channels_latents / 2)
116
+ image = Image.open(image).convert('RGB')
117
  image.resize((size,size), Image.LANCZOS)
118
  with torch.no_grad():
119
  image = pipe.video_processor.preprocess(image, height=size, width=size).to(