Skip to content

Commit d628866

Browse files
remove useless stuff (#58)
1 parent dce6632 commit d628866

File tree

3 files changed

+1
-106
lines changed

3 files changed

+1
-106
lines changed

llmserve/backend/server/models.py

Lines changed: 1 addition & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -303,14 +303,6 @@ def allowed_pipelines(self) -> Set[str]:
303303
return {"vllm"}
304304

305305

306-
class Finetune(Transformers):
307-
type: Literal["Finetune"]
308-
309-
310-
class AutoModel(Transformers):
311-
type: Literal["AutoModel"]
312-
313-
314306
class S3MirrorConfig(BaseModelExtended):
315307
endpoint_url: Optional[str] = None
316308
bucket_uri: Optional[str] = None
@@ -320,7 +312,7 @@ class S3MirrorConfig(BaseModelExtended):
320312

321313
class InitializationConfig(BaseModelExtended):
322314
initializer: Annotated[
323-
Union[DeepSpeed, DeviceMap, SingleDevice, Finetune, AutoModel,
315+
Union[DeepSpeed, DeviceMap, SingleDevice,
324316
LlamaCpp, Vllm], Field(discriminator="type")
325317
]
326318
pipeline: Union[Literal["default"], Literal["defaulttransformers"],
@@ -356,7 +348,6 @@ class GenerationConfig(BaseModelExtended):
356348
}
357349
stopping_sequences: Optional[List[Union[str,
358350
int, List[Union[str, int]]]]] = None
359-
# add_special_tokens: Dict[str, Any] = None
360351

361352
@validator("prompt_format")
362353
def check_prompt_format(cls, value): # pylint:disable=no-self-argument
@@ -382,10 +373,6 @@ def try_int(x):
382373

383374
@property
384375
def all_generate_kwargs(self) -> Dict[str, Any]:
385-
# if self.add_special_tokens:
386-
# return {"stopping_sequences": self.stopping_sequences, **self.generate_kwargs, "add_special_tokens": self.add_special_tokens}
387-
# else:
388-
# return {"stopping_sequences": self.stopping_sequences, **self.generate_kwargs}
389376
return {"stopping_sequences": self.stopping_sequences, **self.generate_kwargs}
390377

391378

models/text-generation--facebook--opt-125m-pipeline.yaml

Lines changed: 0 additions & 50 deletions
This file was deleted.

models/text-generation--facebook--opt-125m-vllm.yaml

Lines changed: 0 additions & 42 deletions
This file was deleted.

0 commit comments

Comments
 (0)