| # -*- coding: utf-8 -*- | |
| # @Function: LHM++ model registry for HuggingFace | |
| HuggingFace_Prior_MODEL_CARD = { | |
| "LHMPP": "3DAIGC/LHMPP-Prior", | |
| } | |
| HuggingFace_MODEL_CARD = { | |
| "LHMPP-700M": "3DAIGC/LHMPP-700M", | |
| # "LHMPP-700MC": "3DAIGC/LHMPP-700MC", # coming soon | |
| "LHMPPS-700M": "3DAIGC/LHMPPS-700M", | |
| } | |
| MODEL_CONFIG = { | |
| "LHMPP-700M": "./configs/train/LHMPP-any-view.yaml", | |
| # "LHMPP-700MC": "./configs/train/LHMPP-any-view-convhead.yaml", # coming soon | |
| "LHMPPS-700M": "./configs/train/LHMPP-any-view-DPTS.yaml", | |
| } | |
| MEMORY_MODEL_CARD = { | |
| "LHMPP-700M": 8000, # 8G | |
| # "LHMPP-700MC": 8000, # 8G, coming soon | |
| "LHMPPS-700M": 8000, # 8G | |
| } |