Buckets:
| from accelerate import Accelerator | |
| from diffusers.utils.torch_utils import is_compiled_module | |
| global_accelerator = None | |
| def get_accelerator() -> Accelerator: | |
| global global_accelerator | |
| if global_accelerator is None: | |
| global_accelerator = Accelerator() | |
| return global_accelerator | |
| def unwrap_model(model): | |
| try: | |
| accelerator = get_accelerator() | |
| model = accelerator.unwrap_model(model) | |
| model = model._orig_mod if is_compiled_module(model) else model | |
| except Exception as e: | |
| pass | |
| return model | |
Xet Storage Details
- Size:
- 551 Bytes
- Xet hash:
- 86137cf63232ddd20d77508bc49f53befeff8b7650424e13c43c0430d2b1f46e
·
Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.