(name)
| 794 | return (patcher,) |
| 795 | |
| 796 | def rename_fuser_block(name): |
| 797 | # map fuser blocks to main blocks |
| 798 | new_name = name |
| 799 | if "face_adapter.fuser_blocks." in name: |
| 800 | match = re.search(r'face_adapter\.fuser_blocks\.(\d+)\.', name) |
| 801 | if match: |
| 802 | fuser_block_num = int(match.group(1)) |
| 803 | main_block_num = fuser_block_num * 5 |
| 804 | new_name = name.replace(f"face_adapter.fuser_blocks.{fuser_block_num}.", f"blocks.{main_block_num}.fuser_block.") |
| 805 | return new_name |
| 806 | |
| 807 | def load_weights(transformer, sd=None, weight_dtype=None, base_dtype=None, |
| 808 | transformer_load_device=None, block_swap_args=None, gguf=False, reader=None, patcher=None, compile_args=None): |
no outgoing calls
no test coverage detected