meccatronis's picture
Upload src/wrapper_for_mps/__init__.py with huggingface_hub
f8c936b verified
Raw History Blame Contribute Delete
239 Bytes
import torch
from comfy.model_management import get_torch_device
device = get_torch_device()
#https://github.com/microsoft/DirectML/issues/414#issuecomment-1541319479
def sparse_to_dense(sparse_tensor):
return sparse_tensor.to_dense()