myniu commited on
Commit
98ab4bc
1 Parent(s): f9cae6d
Files changed (1) hide show
  1. app.py +0 -7
app.py CHANGED
@@ -630,13 +630,6 @@ with gr.Blocks() as demo:
630
 
631
  target_size = 512
632
  DragNUWA_net = Drag(target_size, target_size)
633
-
634
- DragNUWA_net.init_models(
635
- "ckpts/stable-video-diffusion-img2vid-xt-1-1",
636
- "ckpts/controlnet",
637
- weight_dtype=torch.float16,
638
- device='cuda'
639
- )
640
 
641
  first_frame_path = gr.State()
642
  tracking_points = gr.State([])
 
630
 
631
  target_size = 512
632
  DragNUWA_net = Drag(target_size, target_size)
 
 
 
 
 
 
 
633
 
634
  first_frame_path = gr.State()
635
  tracking_points = gr.State([])