Hugging Face's logo

Spaces:
rewgwrth
/
Game-Engine-3-LLM
Sleeping

App Files Community
rewgwrth commited on
Commit
04dec4e
·
1 Parent(s): c31212e

ge3

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -15,7 +15,7 @@ class RequestData(BaseModel):
15
 
16
 
17
  # MODEL_NAME = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
18
- MODEL_NAME = "distilgpt2"
19
 
20
  try:
21
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 
15
 
16
 
17
  # MODEL_NAME = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
18
+ MODEL_NAME = "gpt2"
19
 
20
  try:
21
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)