8 lines
318 B
Python
8 lines
318 B
Python
from pathlib import Path
|
|
|
|
MODEL_DIR = Path(__file__).resolve().parent.parent / "models"
|
|
|
|
CONV_FRONTEND = MODEL_DIR / "model_1.7B" / "conv_frontend.onnx"
|
|
ENCODER = MODEL_DIR / "model_1.7B" / "encoder.int8.onnx"
|
|
DECODER = MODEL_DIR / "model_1.7B" / "decoder.int8.onnx"
|
|
TOKENIZER = MODEL_DIR / "tokenizer" |