| { | |
| "ACTIVATION": "gelu", | |
| "BETA_SCHEDULER": "cosine", | |
| "CFG_WEIGHT": 3, | |
| "CFG_WEIGHT_INDIVIDUAL": 1, | |
| "CFG_WEIGHT_INTERACTION": 3, | |
| "CONTROL": "text", | |
| "DIFFUSION_STEPS": 1000, | |
| "DROPOUT": 0.1, | |
| "FF_SIZE": 2048, | |
| "FINETUNE": false, | |
| "INPUT_DIM": 262, | |
| "LATENT_DIM": 1024, | |
| "MODE": "interaction", | |
| "MOTION_REP": "global", | |
| "NUM_HEADS": 8, | |
| "NUM_LAYERS": 8, | |
| "SAMPLER": "uniform", | |
| "STRATEGY": "ddim50", | |
| "TEXT_ENCODER": "clip", | |
| "T_BAR": 700, | |
| "architectures": [ | |
| "in2INModel" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "config.in2INConfig", | |
| "AutoModel": "model.in2INModel" | |
| }, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.41.2" | |
| } | |