(args)
| 332 | |
| 333 | |
| 334 | def main(args): |
| 335 | model, _, _ = load_model_and_preprocess("blip_diffusion", "base", device="cpu", is_eval=True) |
| 336 | save_blip_diffusion_model(model.state_dict(), args) |
| 337 | |
| 338 | |
| 339 | if __name__ == "__main__": |
no test coverage detected
searching dependent graphs…