diff --git a/llmfoundry/data/finetuning/dataloader.py b/llmfoundry/data/finetuning/dataloader.py index 639beba6f0..0769832962 100644 --- a/llmfoundry/data/finetuning/dataloader.py +++ b/llmfoundry/data/finetuning/dataloader.py @@ -542,7 +542,7 @@ def _download_remote_hf_dataset(remote_path: str, split: str) -> str: except FileNotFoundError as e: if extension == SUPPORTED_EXTENSIONS[-1]: files_searched = [ - f'{name}/{split}{ext}' for ext in SUPPORTED_EXTENSIONS + f'{name[:-len(extension)]}{ext}' for ext in SUPPORTED_EXTENSIONS ] raise FileNotFoundError( f'Could not find a file with any of ' + \