From 68c672d4381dfee21b15e682aba5801bdc3934a8 Mon Sep 17 00:00:00 2001 From: Yizhou Chi Date: Wed, 25 Sep 2024 09:59:12 +0800 Subject: [PATCH] use transformers lib instead of torchvision --- expo/data/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/expo/data/dataset.py b/expo/data/dataset.py index 8ad6f2854..f2f01f71b 100644 --- a/expo/data/dataset.py +++ b/expo/data/dataset.py @@ -26,7 +26,7 @@ TEXT_MODALITY = """ """ IMAGE_MODALITY = """ -7. You could use models from torchvision library for this image dataset. +7. You could use models from transformers library for this image dataset. 8. Use gpu if available for faster training. """