Closed Atomic-Germ closed 1 year ago
Adds a small note to change torch_dtype=torch.float16 to torch_dtype=torch.float32 in the case of a Mac M1 (and related).
To be included in #146, "fixes" #125
Adds a small note to change torch_dtype=torch.float16 to torch_dtype=torch.float32 in the case of a Mac M1 (and related).
To be included in #146, "fixes" #125