Mara Robin B. 1b8f5183c0 load sam_hq to cpu if cuda is unavailable ...

fixes RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

2025-01-05 07:42:14 +01:00

..

modeling

first commit

2024-12-05 17:11:26 +08:00

automatic.py

first commit

2024-12-05 17:11:26 +08:00

build_sam_hq.py

load sam_hq to cpu if cuda is unavailable

2025-01-05 07:42:14 +01:00

predictor.py

first commit

2024-12-05 17:11:26 +08:00