()
| 170 | return model_arch |
| 171 | |
| 172 | def parse_args(): |
| 173 | parser = argparse.ArgumentParser(description="Generate F16 GGUF files from single UNET") |
| 174 | parser.add_argument("--src", required=True, help="Source model ckpt file.") |
| 175 | parser.add_argument("--dst", help="Output unet gguf file.") |
| 176 | args = parser.parse_args() |
| 177 | |
| 178 | if not os.path.isfile(args.src): |
| 179 | parser.error("No input provided!") |
| 180 | |
| 181 | return args |
| 182 | |
| 183 | def strip_prefix(state_dict): |
| 184 | # prefix for mixed state dict |