Zhouyan248 commited on
Commit
b97495d
β€’
1 Parent(s): f5cd544

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -63,7 +63,6 @@ def infer(prompt, image_inp, seed_inp, ddim_steps):
63
  video_input = video_input.to(device).unsqueeze(0)
64
  mask = mask_generation_before(args.mask_type, video_input.shape, video_input.dtype, device)
65
  masked_video = video_input * (mask == 0)
66
- prompt = "tilt up, high quality, stable "
67
  prompt = prompt + args.additional_prompt
68
  video_clip = auto_inpainting(args, video_input, masked_video, mask, prompt, vae, text_encoder, diffusion, model, device,)
69
  video_ = ((video_clip * 0.5 + 0.5) * 255).add_(0.5).clamp_(0, 255).to(dtype=torch.uint8).cpu().permute(0, 2, 3, 1)
 
63
  video_input = video_input.to(device).unsqueeze(0)
64
  mask = mask_generation_before(args.mask_type, video_input.shape, video_input.dtype, device)
65
  masked_video = video_input * (mask == 0)
 
66
  prompt = prompt + args.additional_prompt
67
  video_clip = auto_inpainting(args, video_input, masked_video, mask, prompt, vae, text_encoder, diffusion, model, device,)
68
  video_ = ((video_clip * 0.5 + 0.5) * 255).add_(0.5).clamp_(0, 255).to(dtype=torch.uint8).cpu().permute(0, 2, 3, 1)