Spaces:
Sleeping
Sleeping
danidanidani
commited on
Commit
•
6057e5a
1
Parent(s):
e1d39a6
Update src/backend/chatbot.py
Browse files- src/backend/chatbot.py +0 -5
src/backend/chatbot.py
CHANGED
@@ -33,11 +33,6 @@ print("BP 4 ")
|
|
33 |
def init_llm(model, demo_lite):
|
34 |
# st.write("BP 4.1: model: ", model)
|
35 |
if demo_lite == False:
|
36 |
-
api = HfApi(token=HF_TOKEN)
|
37 |
-
runtime = api.get_space_runtime(repo_id=TRAINING_SPACE_ID)
|
38 |
-
# Check if Space is loaded with a GPU.
|
39 |
-
if runtime.hardware == SpaceHardware.T4_SMALL:
|
40 |
-
print("using nvidia T4 small")
|
41 |
|
42 |
print("BP 5 : running full demo")
|
43 |
if model == "Llama2-7b_CPP":
|
|
|
33 |
def init_llm(model, demo_lite):
|
34 |
# st.write("BP 4.1: model: ", model)
|
35 |
if demo_lite == False:
|
|
|
|
|
|
|
|
|
|
|
36 |
|
37 |
print("BP 5 : running full demo")
|
38 |
if model == "Llama2-7b_CPP":
|