AttributeError: module 'torch._C' has no attribute '_cuda_resetPeakMemoryStats'

54rt1n / ComfyUI-DareMerge

ComfyUI powertools for SD1.5 and SDXL model merging

GNU General Public License v3.0

68 stars 10 forks source link

AttributeError: module 'torch._C' has no attribute '_cuda_resetPeakMemoryStats' #4

Closed a-ru2016 closed 9 months ago

a-ru2016 commented 9 months ago

I'm using an M1macbook, but I got an error because I don't have cuda. /ComfyUI-DareMerge/components/dare_mbw.py", line 86, in merge with cuda_memory_profiler(): It worked except for with cuda_memory_profiler():, so is it possible to change it to turn off this depending on the device?

54rt1n commented 9 months ago

Oof, yes I'll add a check.

54rt1n commented 9 months ago

I pushed a fix, but I can't test it. Let me know if it worked for you.

a-ru2016 commented 9 months ago

It's done! thank you.