Hugging Face's logo

Spaces:
Chris4K
/
A.I.StateMachine
Build error

App Files Community
Chris4K commited on
Commit
7a3dd67
·
verified ·
1 Parent(s): fff699d

Update services/model_service.py

Browse files
Files changed (1) hide show
  1. services/model_service.py +1 -1
services/model_service.py CHANGED
@@ -1,5 +1,5 @@
1
  # services/model_service.py
2
- from transformers import AutoTokenizer, AutoModelForCausalLM
3
  from sentence_transformers import SentenceTransformer
4
  import torch
5
  from functools import lru_cache
 
1
  # services/model_service.py
2
+ from transformers import AutoTokenizer, AutoModelForCausalLM, LlamaConfig
3
  from sentence_transformers import SentenceTransformer
4
  import torch
5
  from functools import lru_cache