| 32 | ) |
| 33 | |
| 34 | func MockASampleModel() *coreapi.OpenModel { |
| 35 | return wrapper.MakeModel(sampleModelName).FamilyName("llama3"). |
| 36 | ModelSourceWithModelHub("Huggingface"). |
| 37 | ModelSourceWithModelID("meta-llama/Meta-Llama-3-8B", "", "", nil, nil). |
| 38 | InferenceFlavors( |
| 39 | *wrapper.MakeFlavor("a100").SetRequest("nvidia.com/gpu", "1").Obj(), |
| 40 | *wrapper.MakeFlavor("a10").SetRequest("nvidia.com/gpu", "2").Obj()). |
| 41 | Obj() |
| 42 | } |
| 43 | |
| 44 | func MockASamplePlayground(ns string) *inferenceapi.Playground { |
| 45 | return wrapper.MakePlayground("playground-llama3-8b", ns).ModelClaim(sampleModelName).Label(coreapi.ModelNameLabelKey, sampleModelName).Obj() |