Muhammad Taqi Raza
commited on
Commit
·
1aab056
1
Parent(s):
ce70d27
GPU setting
Browse files- gradio_app.py +2 -3
gradio_app.py
CHANGED
|
@@ -5,7 +5,7 @@ from pathlib import Path
|
|
| 5 |
import gradio as gr
|
| 6 |
import numpy as np
|
| 7 |
import os
|
| 8 |
-
|
| 9 |
# -----------------------------
|
| 10 |
# Setup paths and env
|
| 11 |
# -----------------------------
|
|
@@ -40,7 +40,6 @@ def download_models():
|
|
| 40 |
# -----------------------------
|
| 41 |
# Step 1: Get Anchor Video
|
| 42 |
# -----------------------------
|
| 43 |
-
@spaces.GPU
|
| 44 |
def get_anchor_video(video_path, fps, num_frames, target_pose, mode,
|
| 45 |
radius_scale, near_far_estimated,
|
| 46 |
sampler_name, diffusion_guidance_scale, diffusion_inference_steps,
|
|
@@ -112,7 +111,7 @@ def get_anchor_video(video_path, fps, num_frames, target_pose, mode,
|
|
| 112 |
# -----------------------------
|
| 113 |
# Step 2: Run Inference
|
| 114 |
# -----------------------------
|
| 115 |
-
@
|
| 116 |
def inference(
|
| 117 |
fps, num_frames, controlnet_weights, controlnet_guidance_start,
|
| 118 |
controlnet_guidance_end, guidance_scale, num_inference_steps, dtype,
|
|
|
|
| 5 |
import gradio as gr
|
| 6 |
import numpy as np
|
| 7 |
import os
|
| 8 |
+
from spaces import GPU
|
| 9 |
# -----------------------------
|
| 10 |
# Setup paths and env
|
| 11 |
# -----------------------------
|
|
|
|
| 40 |
# -----------------------------
|
| 41 |
# Step 1: Get Anchor Video
|
| 42 |
# -----------------------------
|
|
|
|
| 43 |
def get_anchor_video(video_path, fps, num_frames, target_pose, mode,
|
| 44 |
radius_scale, near_far_estimated,
|
| 45 |
sampler_name, diffusion_guidance_scale, diffusion_inference_steps,
|
|
|
|
| 111 |
# -----------------------------
|
| 112 |
# Step 2: Run Inference
|
| 113 |
# -----------------------------
|
| 114 |
+
@GPU
|
| 115 |
def inference(
|
| 116 |
fps, num_frames, controlnet_weights, controlnet_guidance_start,
|
| 117 |
controlnet_guidance_end, guidance_scale, num_inference_steps, dtype,
|