abertsch72 / abertsch72/unlimiformer

Use of other Encode/Decoder Models

Aperta
#55 8 commenti 0 reazioni 0 assegnatari Vedi su GitHub
Lingua principale
Python
Stelle
1.1k
Fork
78
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

Hello, I've been using Unlimiformer as a comparison with current standard methods of summarization and was wondering if there was anything in particular that would be needed in order to convert say a Pegasus model into Unlimiformer as it should work with "All Encoder/Decoder" models. I see several lines commented out in `unlimiformer.py` ([here](https://github.com/abertsch72/unlimiformer/blob/651c5b37d96d676e1da32e36b05dc388bcc440e4/src/unlimiformer.py#L795C8-L802C75)) for AutoModelForSeq2Seq, however I currently dont see a direct way this has been implemented yet.

As Pegasus is BART based, I set up a new model conveter `PegasusForConditionalGeneration: UnlimiformerPegasus,` and started a new unlimiformer class for it:

```
class UnlimiformerPegasus(UnlimiformerBART):
def __init__(self, model: PegasusModel, *args, **kwargs):
super().__init__(model, *args, **kwargs)
```

However, I was wondering if you or anyone else had found additional tweeking that was needed to fully convert say a pegasus model.

And I guess more generally, what is the procedure that you use when setting up your own new unlimiformer converted models as I was unable to simply glean what was necessary to assure "consistent" performance and or results.

Thanks!

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.