| 490 | } |
| 491 | |
| 492 | func (s *SchedulerServer) LoadModel(ctx context.Context, req *pb.LoadModelRequest) (*pb.LoadModelResponse, error) { |
| 493 | logger := s.logger.WithField("func", "LoadModel") |
| 494 | logger.Debugf("Load model %+v k8s meta %+v", req.GetModel().GetMeta(), req.GetModel().GetMeta().GetKubernetesMeta()) |
| 495 | err := s.modelStore.UpdateModel(req) |
| 496 | if err != nil { |
| 497 | return nil, status.Errorf(codes.FailedPrecondition, "%s", err.Error()) |
| 498 | } |
| 499 | go func() { |
| 500 | err := s.scheduler.Schedule(req.GetModel().GetMeta().GetName()) |
| 501 | if err != nil { |
| 502 | logger.WithError(err).Warnf("Failed to schedule model %s", req.GetModel().GetMeta().GetName()) |
| 503 | } |
| 504 | }() |
| 505 | return &pb.LoadModelResponse{}, nil |
| 506 | } |
| 507 | |
| 508 | func (s *SchedulerServer) UnloadModel(ctx context.Context, req *pb.UnloadModelRequest) (*pb.UnloadModelResponse, error) { |
| 509 | logger := s.logger.WithField("func", "UnloadModel") |