Hugging Face's logo

Spaces:
BeveledCube
/
bevelapi
Running

App Files Community
BeveledCube commited on
Commit
1d91f3d
·
verified ·
1 Parent(s): 8fddcc1

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -9,7 +9,7 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer, AutoModelForCausalLM, A
9
  import torch
10
 
11
  app = FastAPI()
12
- name = "microsoft/phi-1_5"
13
  customGen = False
14
 
15
  # microsoft/DialoGPT-small
 
9
  import torch
10
 
11
  app = FastAPI()
12
+ name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
13
  customGen = False
14
 
15
  # microsoft/DialoGPT-small