(lora_weights, hf_modules)
| 200 | |
| 201 | |
| 202 | def get_hf_target_modules(lora_weights, hf_modules): |
| 203 | def iter_fn(layer_idx, hf_module, expert_idx, inout, weights): |
| 204 | hf_target_modules.add(hf_module) |
| 205 | |
| 206 | hf_target_modules = set() |
| 207 | iterate_hf_lora(iter_fn, lora_weights, hf_modules) |
| 208 | return hf_target_modules |
| 209 | |
| 210 | |
| 211 | def invert_module_mapping( |
no test coverage detected