revision not optional, brainforge standard

NeonGeckoCom · Dec 28, 2024 · 016750c · 016750c
1 parent de433e4
commit 016750c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neon_llm_vllm/vllm.py b/neon_llm_vllm/vllm.py
@@ -33,7 +33,7 @@
 from transformers import AutoTokenizer, PreTrainedTokenizerBase
 import numpy as np
 
-from typing import List, Dict, Tuple, Optional
+from typing import List, Dict
 from neon_llm_core.llm import NeonLLM
 from ovos_utils.log import LOG
 
@@ -71,7 +71,7 @@ def from_yaml(cls, yaml_file="datasets/config.yaml"):
 class ModelMetadata:
     vllm_model_name: str
     model_name: str
-    revision: Optional[str] = None
+    revision: str
 
     personas: Dict[str, str]