Spaces:
Running
on
Zero
Running
on
Zero
myniu
commited on
Commit
•
98ab4bc
1
Parent(s):
f9cae6d
init
Browse files
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([])
|