update reqs, change some defaults
This commit is contained in:
@@ -22,7 +22,6 @@ class HumanML3DDataModule(BASEDataModule):
|
||||
|
||||
# Path to the dataset
|
||||
data_root = cfg.DATASET.HUMANML3D.ROOT
|
||||
print(data_root)
|
||||
assets_root = cfg.DATASET.HUMANML3D.ASSETS_ROOT
|
||||
self.hparams.data_root = data_root
|
||||
self.hparams.text_dir = pjoin(data_root, "texts")
|
||||
|
||||
+2
-2
@@ -48,9 +48,9 @@ class mgpt_t2m:
|
||||
def INPUT_TYPES(s):
|
||||
return {"required": {
|
||||
"mgpt_model": ("MGPTMODEL",),
|
||||
"motion_length": ("INT", {"default": 16, "min": 1, "max": 100, "step": 1}),
|
||||
"motion_length": ("INT", {"default": 196, "min": 1, "max": 196, "step": 1}),
|
||||
"seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}),
|
||||
"text": ("STRING", {"default": 'jump'}),
|
||||
"text": ("STRING", {"multiline": True,"default": 'make the person jump and turn around'}),
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
+3
-1
@@ -19,4 +19,6 @@ mapbox_earcut
|
||||
spacy
|
||||
bert-score
|
||||
omegaconf
|
||||
huggingface_hub
|
||||
huggingface_hub
|
||||
pytorch_lightning
|
||||
rich
|
||||
+8
-8
@@ -26,7 +26,7 @@ class SmplifyMotionData:
|
||||
return {
|
||||
"required": {
|
||||
"motion_data": ("MOTION_DATA", ),
|
||||
"num_smplify_iters": ("INT", {"min": 1, "max": 1000, "default": 50}),
|
||||
"num_smplify_iters": ("INT", {"min": 1, "max": 1000, "default": 20}),
|
||||
"smplify_step_size": ("FLOAT", {"min": 1e-4, "max": 5e-1, "step": 1e-4, "default": 1e-1}),
|
||||
"smpl_model": (list(smpl_model_dicts.keys()), {"default": "SMPL_NEUTRAL.pkl"})
|
||||
}
|
||||
@@ -88,16 +88,16 @@ class RenderSMPLMesh:
|
||||
"smpl": ("SMPL", ),
|
||||
"draw_platform": ("BOOLEAN", {"default": False}),
|
||||
"depth_only": ("BOOLEAN", {"default": False}),
|
||||
"yfov": ("FLOAT", {"default": 0.75, "min": 0.1, "max": 10, "step": 0.01}),
|
||||
"move_x": ("FLOAT", {"default": 0,"min": -500, "max": 500, "step": 0.1}),
|
||||
"move_y": ("FLOAT", {"default": 0,"min": -500, "max": 500, "step": 0.1}),
|
||||
"move_z": ("FLOAT", {"default": 0,"min": -500, "max": 500, "step": 0.1}),
|
||||
"yfov": ("FLOAT", {"default": 0.6, "min": 0.1, "max": 10, "step": 0.01}),
|
||||
"move_x": ("FLOAT", {"default": 0,"min": -500, "max": 500, "step": 0.01}),
|
||||
"move_y": ("FLOAT", {"default": -0.1,"min": -500, "max": 500, "step": 0.01}),
|
||||
"move_z": ("FLOAT", {"default": 0,"min": -500, "max": 500, "step": 0.01}),
|
||||
"rotate_x": ("FLOAT", {"default": 0,"min": -180, "max": 180, "step": 0.1}),
|
||||
"rotate_y": ("FLOAT", {"default": 0,"min": -180, "max": 180, "step": 0.1}),
|
||||
"rotate_z": ("FLOAT", {"default": 0,"min": -180, "max": 180, "step": 0.1}),
|
||||
"background_hex_color": ("STRING", {"default": "#FFFFFF", "mutiline": False}),
|
||||
"frame_width": ("INT", {"default": 960, "min": 0, "max": 4096, "step": 1}),
|
||||
"frame_height": ("INT", {"default": 960, "min": 0, "max": 4096, "step": 1}),
|
||||
"background_hex_color": ("STRING", {"default": "#000000", "mutiline": False}),
|
||||
"frame_width": ("INT", {"default": 512, "min": 64, "max": 4096, "step": 8}),
|
||||
"frame_height": ("INT", {"default": 512, "min": 64, "max": 4096, "step": 8}),
|
||||
},
|
||||
"optional": {
|
||||
"normals": ("BOOLEAN", {"default": False}),
|
||||
|
||||
Reference in New Issue
Block a user