diff --git a/internal/control/modelserve/plan.go b/internal/control/modelserve/plan.go index 58e48f6..9767fbc 100644 --- a/internal/control/modelserve/plan.go +++ b/internal/control/modelserve/plan.go @@ -104,6 +104,10 @@ func (s *Server) plan(ctx context.Context, repo hf.Repo, req PlanRequest) (Plan, p := Plan{Repo: repo.ID, Revision: repo.Revision, Format: "strata", Arch: repo.Architecture, License: repo.License, Context: strataContext, Servable: true, Note: "Strata fetches this pack onto the node itself on its first start"} p.Options = modelplan.StrataOptions(fam.Name, []modelplan.Artifact{art}, s.deps.Pool()) + if len(p.Options) == 0 && fam.MinCC > 0 { + p.Note = fmt.Sprintf("Strata's %s engine needs an NVIDIA card of compute capability %.1f or newer (RTX 30 on); "+ + "a node with Strata and such a card, and %d GB of RAM, runs it", fam.Title, fam.MinCC, size.MemoryGB()) + } pa := PlannedArtifact{Artifact: art} if len(p.Options) > 0 { pa.Best = &p.Options[0]