Update custom_generate/generate.py
#1
by
RaushanTurganbay
HF Staff
- opened
Makes sure that the core generate() gets called which prepares generation config, model kwargs, logits processors and so on. Currently the custom generation fails with some models due to model kwargs being incomplete, e.g. gemma3