(model)
| 934 | flux_controlnet_pipeline.to(accelerator.device) |
| 935 | |
| 936 | def unwrap_model(model): |
| 937 | model = accelerator.unwrap_model(model) |
| 938 | model = model._orig_mod if is_compiled_module(model) else model |
| 939 | return model |
| 940 | |
| 941 | # `accelerate` 0.16.0 will have better support for customized saving |
| 942 | if version.parse(accelerate.__version__) >= version.parse("0.16.0"): |
no test coverage detected