heikowagner commited on
Commit
c62c32d
1 Parent(s): 13ad645

Update app/app.py

Browse files
Files changed (1) hide show
  1. 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< 17254768640:
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")