 
         // Downloads and extracts dataset files.
         let _ = DatasetUtilities.downloadResource(
-            filename: variantDetails.filename, fileExtension: variantDetails.fileExtension,
+            filename: (downloadEncodedFile ? variantDetails.encodedFileName! : variantDetails.filename),
+            fileExtension: variantDetails.fileExtension,
             remoteRoot: variantDetails.location, localStorageDirectory: directory, extract: true)
     }
 

    var trainingDirectoryName: String { get set }
    var validationDirectoryName: String { get set }
    var filename: String { get set }
    var fileExtension: String { get set }
}


    }

    public init(
        bpe: BytePairEncoder? = nil,
        variant: TextUnsupervisedVariant = TextUnsupervisedVariant.wikiText2,
        trainingBatchSize: Int = 8, validationBatchSize: Int = 4, sequenceLength: Int = 1024,
        trainingDocumentCount: Int = 4, validationDocumentCount: Int = 4
