diff --git a/hft.py b/hft.py
index 1a679ce..ef4bfeb 100644
--- a/hft.py
+++ b/hft.py
@@ -143,7 +143,7 @@ class CustomModel(GPTNeoForCausalLM):
         return outputs
 
 source_mapper = SourceMapper()
-model_name = "EleutherAI/gpt-neo-2.7B"
+model_name = "EleutherAI/gpt-neo-1.3B"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 tokenizer.pad_token = tokenizer.eos_token