diff --git a/fmperf/__init__.py b/fmperf/__init__.py index c629998..520d38f 100644 --- a/fmperf/__init__.py +++ b/fmperf/__init__.py @@ -1,3 +1,3 @@ from fmperf.ModelSpecs import TGISModelSpec, vLLMModelSpec from fmperf.Cluster import Cluster, WorkloadSpec, DeployedModel -from fmperf.WorkloadSpecs import HeterogeneousWorkloadSpec, HomogeneousWorkloadSpec +from fmperf.WorkloadSpecs import HeterogeneousWorkloadSpec, HomogeneousWorkloadSpec, RealisticWorkloadSpec