Neuracraft / model_loading.py
Erfan11's picture
Update model_loading.py
6f9e18f verified
raw
history blame
No virus
144 Bytes
from load_model import load_model
model, tokenizer = load_model()
def get_model():
return model
def get_tokenizer():
return tokenizer