Commit e6d8ad8
Resolve mappings for Cohere2ForCausalLM, Glm4MoeForCausalLM, SeedOssForCausalLM, and Ernie4_5_MoeForCausalLM (#1926)
This repo is to resolve mappings for Cohere2ForCausalLM,
Glm4MoeForCausalLM, SeedOssForCausalLM, and Ernie4_5_MoeForCausalLM
models.
TEST PLAN:
Local make test results:
```
======================================================================== short test summary info =========================================================================
FAILED tests/llmcompressor/modeling/test_calib_deepseek_v3.py::test_calib_deepseekv3_module - torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 56.00 MiB. GPU 0 has a total capacity of 23.57 GiB of which 14.12 MiB is free. Including non-PyTorch mem...
FAILED tests/llmcompressor/utils/test_helpers.py::test_disable_cache[MllamaForConditionalGeneration-meta-llama/Llama-3.2-11B-Vision-Instruct] - torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 64.00 MiB. GPU 0 has a total capacity of 23.57 GiB of which 54.12 MiB is free. Including non-PyTorch mem...
FAILED tests/lmeval/test_lmeval.py::TestLMEval::test_lm_eval[None] - TypeError: argument should be a str or an os.PathLike object where __fspath__ returns a str, not 'NoneType'
=========================================================== 3 failed, 242 passed, 4 skipped in 80.83s (0:01:20) ===========================================================
```
Co-authored-by: toncao <[email protected]>
Co-authored-by: Brian Dellabetta <[email protected]>1 parent 7194d4b commit e6d8ad8
1 file changed
+5
-0
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
147 | 147 | | |
148 | 148 | | |
149 | 149 | | |
| 150 | + | |
150 | 151 | | |
151 | 152 | | |
152 | 153 | | |
153 | 154 | | |
154 | 155 | | |
| 156 | + | |
155 | 157 | | |
156 | 158 | | |
157 | 159 | | |
| |||
161 | 163 | | |
162 | 164 | | |
163 | 165 | | |
| 166 | + | |
| 167 | + | |
| 168 | + | |
164 | 169 | | |
165 | 170 | | |
166 | 171 | | |
| |||
0 commit comments