Isa0
/

text-to-speech-train-code

Isa0 commited on Jun 8

Commit

378dec0

1 Parent(s): e498f5b

fix runtime error with trust_remote_code=True

Files changed (1) hide show

train.py CHANGED Viewed

@@ -66,7 +66,7 @@ if __name__ == "__main__":
     tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
     processor = Wav2Vec2Processor.from_pretrained(MODEL_ID)
-    dataset = load_dataset(DATASET_ID, split="train")
     dataset = dataset.train_test_split(test_size=0.05, seed=42)
     dataset = dataset.map(

     tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
     processor = Wav2Vec2Processor.from_pretrained(MODEL_ID)
+    dataset = load_dataset(DATASET_ID, split="train", trust_remote_code=True)
     dataset = dataset.train_test_split(test_size=0.05, seed=42)
     dataset = dataset.map(