Hugging Face's logo

Spaces:
helloperson123
/
tiny-llama-chatbot
Runtime error

App Files Community
helloperson123 commited on
Commit
67abc83
·
verified ·
1 Parent(s): 7dee2e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ CORS(app)
10
  # ----------------------------
11
  # Model (UNGATED, FREE)
12
  # ----------------------------
13
- MODEL_NAME = "microsoft/phi-2"
14
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
15
 
16
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 
10
  # ----------------------------
11
  # Model (UNGATED, FREE)
12
  # ----------------------------
13
+ MODEL_NAME = "microsoft/phi-3"
14
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
15
 
16
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)