Using device: mps
Traceback (most recent call last):
File "Disco_Diffusion_v5_2_m1.py", line 2340, in <module>
File "Disco_Diffusion_v5_2_m1.py", line 983, in do_run
txt = clip_model.encode_text(clip.tokenize(prompt).to(device)).float()
File "/Users/aiden/notebook/CLIP/clip/model.py", line 355, in encode_text
x = x[torch.arange(x.shape), text.argmax(dim=-1)] @ self.text_projection
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)
NotImplementedError: The operator 'aten::index.Tensor_out' is not current implemented for the MPS device. If you want this op to be added in priority during the prototype phase of this feature, please comment on https://github.com/pytorch/pytorch/issues/77764. As a temporary fix, you can set the environment variable `PYTORCH_ENABLE_MPS_FALLBACK=1` to use the CPU as a fallback for this op. WARNING: this will be slower than running natively on MPS.
‘aten::index.Tensor_out’ triggers fallback to cpu.