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