Is the “torbo” class model not supported?

jzbjyb / FLARE

Forward-Looking Active REtrieval-augmented generation (FLARE)

MIT License

579 stars 51 forks source link

Is the “torbo” class model not supported? #18

Open zoeChen119 opened 8 months ago

zoeChen119 commented 8 months ago

'text-davinci-002', 'text-davinci-003' have been deprecated, https://platform.openai.com/docs/deprecations. in templates.py, only torbo model supported? It seems that 'text-davinci-002', 'text-davinci-003' cannot generate token?

def truncate_at_prob(self, low: float):
        assert self.has_tokens, 'not supported'

        if self.num_tokens <= 1:
            return self