fix: patch build_model.py cuda calls + torch.load map_location b957821 Jjtumarai commited on 5 days ago