Spaces:
Running
on
A100
Running
on
A100
Update llava/data/dataset.py
Browse files- llava/data/dataset.py +1 -1
llava/data/dataset.py
CHANGED
@@ -161,7 +161,7 @@ class LazySupervisedDataset(Dataset):
|
|
161 |
self.list_data_dict = list_data_dict
|
162 |
self.data_args = data_args
|
163 |
self.image_folder = image_folder
|
164 |
-
self.wav_processor = AutoFeatureExtractor.from_pretrained('/
|
165 |
|
166 |
def __len__(self):
|
167 |
return len(self.list_data_dict)
|
|
|
161 |
self.list_data_dict = list_data_dict
|
162 |
self.data_args = data_args
|
163 |
self.image_folder = image_folder
|
164 |
+
self.wav_processor = AutoFeatureExtractor.from_pretrained('Qwen/Qwen2-Audio-7B')
|
165 |
|
166 |
def __len__(self):
|
167 |
return len(self.list_data_dict)
|