From b34da419d0f8c1d822f0168bc0bc1acaeece0f7c Mon Sep 17 00:00:00 2001 From: andreapdr Date: Thu, 16 Mar 2023 11:49:49 +0100 Subject: [PATCH] fixed import --- gfun/vgfs/visualTransformerGen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gfun/vgfs/visualTransformerGen.py b/gfun/vgfs/visualTransformerGen.py index c3994e8..c105003 100644 --- a/gfun/vgfs/visualTransformerGen.py +++ b/gfun/vgfs/visualTransformerGen.py @@ -9,7 +9,7 @@ from transformers import AutoImageProcessor, AutoModelForImageClassification from gfun.vgfs.commons import Trainer from gfun.vgfs.transformerGen import TransformerGen from gfun.vgfs.viewGen import ViewGen -from dataManager.torchDataset import MultilingualDatasetTorch +from dataManager.torchDataset import MultimodalDatasetTorch transformers.logging.set_verbosity_error()