kaburia commited on
Commit
0ed976e
·
1 Parent(s): 7139211
Files changed (1) hide show
  1. utils/generation_streaming.py +1 -0
utils/generation_streaming.py CHANGED
@@ -23,6 +23,7 @@ from utils.loading_embeddings import get_vectorstore
23
  # build message from model generation
24
  from utils.model_generation import build_messages
25
  import os
 
26
 
27
  API_KEY = "sk-do-8Hjf0liuGQCoPwglilL49xiqrthMECwjGP_kAjPM53OTOFQczPyfPK8xJc"
28
  MODEL = "llama3.3-70b-instruct"
 
23
  # build message from model generation
24
  from utils.model_generation import build_messages
25
  import os
26
+ import spaces
27
 
28
  API_KEY = "sk-do-8Hjf0liuGQCoPwglilL49xiqrthMECwjGP_kAjPM53OTOFQczPyfPK8xJc"
29
  MODEL = "llama3.3-70b-instruct"