SreyanG-NVIDIA commited on
Commit
564c660
·
verified ·
1 Parent(s): de8e389

Update llava/data/dataset.py

Browse files
Files changed (1) hide show
  1. 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('/lustre/fsw/portfolios/adlr/users/sreyang/flamingo_v2/NV-Whisper')
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)