Hugging Face's logo

Spaces:
nroggendorff
/
train-llama
Runtime error

App Files Community
34
nroggendorff commited on
Commit
9783042
·
verified ·
1 Parent(s): 040f347

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -99,7 +99,7 @@ def create_model(tokenizer):
99
  return model
100
 
101
  def load_model():
102
- model = AutoModelForCasualLM.from_pretrained(OUTPUT_REPO)
103
  return model
104
 
105
  def configure_tokenizer(tokenizer):
 
99
  return model
100
 
101
  def load_model():
102
+ model = AutoModelForCausalLM.from_pretrained(OUTPUT_REPO)
103
  return model
104
 
105
  def configure_tokenizer(tokenizer):