This commit is contained in:
David Kyle 2023-01-25 11:51:47 +00:00
parent 3a044a76e3
commit e9babdcc37

View File

@ -123,6 +123,7 @@ class InferenceConfig:
}
}
class SlimInferenceOptions(InferenceConfig):
def __init__(
self,
@ -134,6 +135,7 @@ class SlimInferenceOptions(InferenceConfig):
self.tokenization = tokenization
self.results_field = results_field
class TextClassificationInferenceOptions(InferenceConfig):
def __init__(
self,