(name)
| 183 | end_layer = min(num_layer, args.rank * part_layer + part_layer) |
| 184 | |
| 185 | def get_tensor(name): |
| 186 | if name not in weight_map: |
| 187 | return None |
| 188 | ff = weight_map[name] |
| 189 | safetensors_loader = processed_files[ff] |
| 190 | return safetensors_loader.get_tensor(name).cuda() |
| 191 | |
| 192 | def get_file_name(layer): |
| 193 | rank = layer // part_layer |