LovnishVerma commited on
Commit
86d0067
1 Parent(s): 5130d64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,6 +3,6 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
3
  model_name = "mlabonne/llama-2-7b-guanaco"
4
 
5
  try:
6
- model = AutoModelForCausalLM.from_pretrained(model_name)
7
  except Exception as e:
8
  print(f"Error loading the model: {e}")
 
3
  model_name = "mlabonne/llama-2-7b-guanaco"
4
 
5
  try:
6
+ model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=False, device='cpu', memory_limit=1024*16)
7
  except Exception as e:
8
  print(f"Error loading the model: {e}")