Attempting to deserialize object on a CUDA

ZHO-ZHO-ZHO / ComfyUI-BiRefNet-ZHO

Better version for BiRefNet in ComfyUI | Both img & video

GNU General Public License v3.0

270 stars 22 forks source link

Attempting to deserialize object on a CUDA #5

Open water2891 opened 7 months ago

water2891 commented 7 months ago

File "/Users/shijin/ComfyUI/custom_nodes/ComfyUI-BiRefNet-ZHO/models/backbones/build_backbone.py", line 29, in load_weights save_model = torch.load(config.weights[model_name])

在CPU模式下，这里报错，缺map_location参数

Bigleeeo commented 4 months ago

我也是

HansonChan commented 3 months ago

自己改了先凑合用 build_backbone.py line29 save_model = torch.load(config.weights[model_name]) save_model = torch.load(config.weights[model_name],map_location=torch.device('mps')) torch.device那cpu的填cpu，mac的填mps