update reqs, change some defaults

This commit is contained in:
kijai
2024-03-29 18:57:28 +02:00
parent e2c67e5abe
commit 691c9e05a4
4 changed files with 13 additions and 12 deletions
-1
View File
@@ -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
View File
@@ -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
View File
@@ -19,4 +19,6 @@ mapbox_earcut
spacy
bert-score
omegaconf
huggingface_hub
huggingface_hub
pytorch_lightning
rich
+8 -8
View File
@@ -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}),