Spaces:

cconsti
/

trial1

Runtime error

cconsti commited on Jan 31

Commit

e02989b

verified ·

1 Parent(s): 49eb74f

Update train.py

Files changed (1) hide show

train.py CHANGED Viewed

@@ -24,7 +24,7 @@ if "test" not in dataset:
 model_name = "t5-large"
 tokenizer = T5Tokenizer.from_pretrained(model_name)
 model = T5ForConditionalGeneration.from_pretrained(model_name)
-model.gradient_checkpointing_enable(False)
 # Define tokenization function
 def tokenize_function(examples):

 model_name = "t5-large"
 tokenizer = T5Tokenizer.from_pretrained(model_name)
 model = T5ForConditionalGeneration.from_pretrained(model_name)
+model.gradient_checkpointing_disable()
 # Define tokenization function
 def tokenize_function(examples):