InnerI
/

synCAI-144k-gpt2.5

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

InnerI commited on May 23

Commit

739de96

•

1 Parent(s): 147b54b

Update chat.py

Files changed (1) hide show

chat.py +1 -1

chat.py CHANGED Viewed

@@ -2,7 +2,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 # Load the tokenizer and model
-model_name = "InnerI/synCAI-144k-gpt2.5"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)

 from transformers import AutoTokenizer, AutoModelForCausalLM
 # Load the tokenizer and model
+model_name = "synCAI-144k-gpt2.5"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)