Muhammad Taqi Raza
commited on
Commit
·
82e411f
1
Parent(s):
062d97b
absolute paths
Browse files
inference/v2v_data/inference.py
CHANGED
|
@@ -84,7 +84,7 @@ def get_parser():
|
|
| 84 |
parser.add_argument(
|
| 85 |
'--model_name',
|
| 86 |
type=str,
|
| 87 |
-
default='
|
| 88 |
help='Path to the model',
|
| 89 |
)
|
| 90 |
parser.add_argument(
|
|
@@ -98,7 +98,7 @@ def get_parser():
|
|
| 98 |
parser.add_argument(
|
| 99 |
'--transformer_path',
|
| 100 |
type=str,
|
| 101 |
-
default="
|
| 102 |
help='Path to the pretrained transformer model',
|
| 103 |
)
|
| 104 |
parser.add_argument(
|
|
@@ -142,7 +142,7 @@ def get_parser():
|
|
| 142 |
parser.add_argument(
|
| 143 |
'--unet_path',
|
| 144 |
type=str,
|
| 145 |
-
default="
|
| 146 |
help='Path to the UNet model',
|
| 147 |
)
|
| 148 |
|
|
@@ -150,7 +150,7 @@ def get_parser():
|
|
| 150 |
parser.add_argument(
|
| 151 |
'--pre_train_path',
|
| 152 |
type=str,
|
| 153 |
-
default="
|
| 154 |
help='Path to the pre-trained model',
|
| 155 |
)
|
| 156 |
parser.add_argument(
|
|
|
|
| 84 |
parser.add_argument(
|
| 85 |
'--model_name',
|
| 86 |
type=str,
|
| 87 |
+
default='/app/pretrained/CogVideoX-Fun-V1.1-5b-InP',
|
| 88 |
help='Path to the model',
|
| 89 |
)
|
| 90 |
parser.add_argument(
|
|
|
|
| 98 |
parser.add_argument(
|
| 99 |
'--transformer_path',
|
| 100 |
type=str,
|
| 101 |
+
default="/app/pretrained/TrajectoryCrafter",
|
| 102 |
help='Path to the pretrained transformer model',
|
| 103 |
)
|
| 104 |
parser.add_argument(
|
|
|
|
| 142 |
parser.add_argument(
|
| 143 |
'--unet_path',
|
| 144 |
type=str,
|
| 145 |
+
default="/app/pretrained/DepthCrafter",
|
| 146 |
help='Path to the UNet model',
|
| 147 |
)
|
| 148 |
|
|
|
|
| 150 |
parser.add_argument(
|
| 151 |
'--pre_train_path',
|
| 152 |
type=str,
|
| 153 |
+
default="/app/pretrained/stable-video-diffusion-img2vid",
|
| 154 |
help='Path to the pre-trained model',
|
| 155 |
)
|
| 156 |
parser.add_argument(
|