llm

Sleeping

Chris4K commited on Jan 12

Commit

7a3dd67

verified ·

1 Parent(s): fff699d

Update services/model_service.py

Files changed (1) hide show

services/model_service.py CHANGED Viewed

@@ -1,5 +1,5 @@
 # services/model_service.py
-from transformers import AutoTokenizer, AutoModelForCausalLM
 from sentence_transformers import SentenceTransformer
 import torch
 from functools import lru_cache

 # services/model_service.py
+from transformers import AutoTokenizer, AutoModelForCausalLM, LlamaConfig
 from sentence_transformers import SentenceTransformer
 import torch
 from functools import lru_cache