jhj0517
commited on
Commit
•
da4dd26
1
Parent(s):
c498b19
set duration to 120
Browse files- musepose_inference.py +1 -1
- pose_align.py +1 -1
musepose_inference.py
CHANGED
|
@@ -50,7 +50,7 @@ class MusePoseInference:
|
|
| 50 |
if not os.path.exists(self.output_dir):
|
| 51 |
os.makedirs(self.output_dir)
|
| 52 |
|
| 53 |
-
@spaces.GPU(duration=
|
| 54 |
def infer_musepose(
|
| 55 |
self,
|
| 56 |
ref_image_path: str,
|
|
|
|
| 50 |
if not os.path.exists(self.output_dir):
|
| 51 |
os.makedirs(self.output_dir)
|
| 52 |
|
| 53 |
+
@spaces.GPU(duration=120)
|
| 54 |
def infer_musepose(
|
| 55 |
self,
|
| 56 |
ref_image_path: str,
|
pose_align.py
CHANGED
|
@@ -39,7 +39,7 @@ class PoseAlignmentInference:
|
|
| 39 |
if not os.path.exists(self.output_dir):
|
| 40 |
os.makedirs(self.output_dir)
|
| 41 |
|
| 42 |
-
@spaces.GPU(duration=
|
| 43 |
def align_pose(
|
| 44 |
self,
|
| 45 |
vidfn: str,
|
|
|
|
| 39 |
if not os.path.exists(self.output_dir):
|
| 40 |
os.makedirs(self.output_dir)
|
| 41 |
|
| 42 |
+
@spaces.GPU(duration=120)
|
| 43 |
def align_pose(
|
| 44 |
self,
|
| 45 |
vidfn: str,
|