Spaces:
Build error
Build error
heikowagner
commited on
Commit
•
c62c32d
1
Parent(s):
13ad645
Update app/app.py
Browse files- app/app.py +1 -1
app/app.py
CHANGED
@@ -36,7 +36,7 @@ else:
|
|
36 |
llm= load_model.load_openai_model()
|
37 |
elif model_type=='decapoda-research/llama-7b-hf (gpu+cpu)':
|
38 |
# Add more models here
|
39 |
-
if not torch.cuda.is_available() and psutil.virtual_memory().available<
|
40 |
st.write('You do not have enough memory to use this model:' + str(psutil.virtual_memory().available))
|
41 |
else:
|
42 |
llm = load_model.load_gpu_model("decapoda-research/llama-7b-hf")
|
|
|
36 |
llm= load_model.load_openai_model()
|
37 |
elif model_type=='decapoda-research/llama-7b-hf (gpu+cpu)':
|
38 |
# Add more models here
|
39 |
+
if not torch.cuda.is_available() and psutil.virtual_memory().available< 18254768640:
|
40 |
st.write('You do not have enough memory to use this model:' + str(psutil.virtual_memory().available))
|
41 |
else:
|
42 |
llm = load_model.load_gpu_model("decapoda-research/llama-7b-hf")
|