search

oriyor / ret-robust

Implementation of the paper: "Making Retrieval-Augmented Language Models Robust to Irrelevant Context"
MIT License
59 stars 1 forks source link

DataCollatorSelfAsk #4

Closed CuteMing closed 2 months ago

CuteMing commented 6 months ago

from data_collator import DataCollatorSelfAsk

ModuleNotFoundError: No module named 'data_collator ImportError: cannot import name 'DataCollatorSelfAsk' from 'transformers.data.data_collator'

This module is not found, can you upload the requirements.txt, or the version of transformers, thanks!!

oriyor commented 2 months ago

Hi, sorry for the late response, please see fix here - https://github.com/oriyor/ret-robust/issues/6