diff --git a/mGPT/data/HumanML3D.py b/mGPT/data/HumanML3D.py index 8bec575..5213627 100644 --- a/mGPT/data/HumanML3D.py +++ b/mGPT/data/HumanML3D.py @@ -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") diff --git a/mgpt_nodes.py b/mgpt_nodes.py index d932626..ceb147c 100644 --- a/mgpt_nodes.py +++ b/mgpt_nodes.py @@ -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'}), } } diff --git a/requirements.txt b/requirements.txt index 7d637e6..d550179 100644 --- a/requirements.txt +++ b/requirements.txt @@ -19,4 +19,6 @@ mapbox_earcut spacy bert-score omegaconf -huggingface_hub \ No newline at end of file +huggingface_hub +pytorch_lightning +rich \ No newline at end of file diff --git a/smpl_nodes.py b/smpl_nodes.py index 9715009..36186c3 100644 --- a/smpl_nodes.py +++ b/smpl_nodes.py @@ -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}),