wiikoo commited on
Commit
3193e48
·
verified ·
1 Parent(s): fcd06ec

Upload custom_nodes/comfyui_controlnet_aux/src/wrapper_for_mps/__init__.py with huggingface_hub

Browse files
custom_nodes/comfyui_controlnet_aux/src/wrapper_for_mps/__init__.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from comfy.model_management import get_torch_device
3
+
4
+ device = get_torch_device()
5
+ #https://github.com/microsoft/DirectML/issues/414#issuecomment-1541319479
6
+ def sparse_to_dense(sparse_tensor):
7
+ return sparse_tensor.to_dense()