Fix PyTorch 2.3.1 compatibility: add version guard for torch.library.… by Aishwarya0811 · Pull Request #12206 · huggingface/diffusers (original) (raw)
| if torch.__version__ >= "2.4.0": |
|---|
| _custom_op = torch.library.custom_op |
| _register_fake = torch.library.register_fake |
| else: |
| def _custom_op_no_op(name, fn=None, /, *, mutates_args, device_types=None, schema=None): |
| def wrap(func): |
| return func |
| return wrap if fn is None else fn |
| def _register_fake_no_op(op, fn=None, /, *, lib=None, _stacklevel=1): |
| def wrap(func): |
| return func |
| return wrap if fn is None else fn |
| _custom_op = _custom_op_no_op |
| _register_fake = _register_fake_no_op |