@@ -303,14 +303,6 @@ def allowed_pipelines(self) -> Set[str]:
303
303
return {"vllm" }
304
304
305
305
306
- class Finetune (Transformers ):
307
- type : Literal ["Finetune" ]
308
-
309
-
310
- class AutoModel (Transformers ):
311
- type : Literal ["AutoModel" ]
312
-
313
-
314
306
class S3MirrorConfig (BaseModelExtended ):
315
307
endpoint_url : Optional [str ] = None
316
308
bucket_uri : Optional [str ] = None
@@ -320,7 +312,7 @@ class S3MirrorConfig(BaseModelExtended):
320
312
321
313
class InitializationConfig (BaseModelExtended ):
322
314
initializer : Annotated [
323
- Union [DeepSpeed , DeviceMap , SingleDevice , Finetune , AutoModel ,
315
+ Union [DeepSpeed , DeviceMap , SingleDevice ,
324
316
LlamaCpp , Vllm ], Field (discriminator = "type" )
325
317
]
326
318
pipeline : Union [Literal ["default" ], Literal ["defaulttransformers" ],
@@ -356,7 +348,6 @@ class GenerationConfig(BaseModelExtended):
356
348
}
357
349
stopping_sequences : Optional [List [Union [str ,
358
350
int , List [Union [str , int ]]]]] = None
359
- # add_special_tokens: Dict[str, Any] = None
360
351
361
352
@validator ("prompt_format" )
362
353
def check_prompt_format (cls , value ): # pylint:disable=no-self-argument
@@ -382,10 +373,6 @@ def try_int(x):
382
373
383
374
@property
384
375
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}
389
376
return {"stopping_sequences" : self .stopping_sequences , ** self .generate_kwargs }
390
377
391
378
0 commit comments