Update services/model_service.py
Browse files
services/model_service.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
from transformers import AutoTokenizer, AutoModelForCausalLM, LlamaConfig
|
|
|
2 |
from sentence_transformers import SentenceTransformer
|
3 |
import torch
|
4 |
import logging
|
|
|
1 |
from transformers import AutoTokenizer, AutoModelForCausalLM, LlamaConfig
|
2 |
+
from config.config import settings
|
3 |
from sentence_transformers import SentenceTransformer
|
4 |
import torch
|
5 |
import logging
|