jhj0517
commited on
Commit
•
af90a1b
1
Parent(s):
7b495d7
download models at start
Browse files- musepose_inference.py +1 -1
- pose_align.py +1 -1
musepose_inference.py
CHANGED
|
@@ -41,7 +41,7 @@ class MusePoseInference:
|
|
| 41 |
self.image_enc = None
|
| 42 |
self.pipe = None
|
| 43 |
self.output_dir = os.path.join("assets", "video")
|
| 44 |
-
|
| 45 |
|
| 46 |
def infer_musepose(
|
| 47 |
self,
|
|
|
|
| 41 |
self.image_enc = None
|
| 42 |
self.pipe = None
|
| 43 |
self.output_dir = os.path.join("assets", "video")
|
| 44 |
+
self.download_models()
|
| 45 |
|
| 46 |
def infer_musepose(
|
| 47 |
self,
|
pose_align.py
CHANGED
|
@@ -31,7 +31,7 @@ class PoseAlignmentInference:
|
|
| 31 |
"det_config": os.path.join("pose", "config", "yolox_l_8xb8-300e_coco.py"),
|
| 32 |
}
|
| 33 |
self.output_dir = os.path.join("assets", "video")
|
| 34 |
-
|
| 35 |
|
| 36 |
def align_pose(
|
| 37 |
self,
|
|
|
|
| 31 |
"det_config": os.path.join("pose", "config", "yolox_l_8xb8-300e_coco.py"),
|
| 32 |
}
|
| 33 |
self.output_dir = os.path.join("assets", "video")
|
| 34 |
+
self.download_models()
|
| 35 |
|
| 36 |
def align_pose(
|
| 37 |
self,
|