Lint

SeldonIO · Jul 28, 2023 · 63a41dc · 63a41dc
1 parent 55635b3
commit 63a41dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/runtimes/huggingface/mlserver_huggingface/common.py b/runtimes/huggingface/mlserver_huggingface/common.py
@@ -63,7 +63,7 @@ def load_pipeline_from_settings(
     # If max_batch_size > 0 we need to ensure tokens are padded
     if settings.max_batch_size:
         model = hf_pipeline.model
-        eos_token_id = model.config.eos_token_id # type: ignore
+        eos_token_id = model.config.eos_token_id  # type: ignore
         hf_pipeline.tokenizer.pad_token_id = [str(eos_token_id)]  # type: ignore
 
     return hf_pipeline