Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Suppress Generate config GenerationConfig #212

Open
francescoboc opened this issue Feb 23, 2023 · 2 comments
Open

Suppress Generate config GenerationConfig #212

francescoboc opened this issue Feb 23, 2023 · 2 comments

Comments

@francescoboc
Copy link

francescoboc commented Feb 23, 2023

Everytime I call ai.generate() I get these lines:

Generate config GenerationConfig {
  "bos_token_id": 0,
  "eos_token_id": 0,
  "transformers_version": "4.26.1"
}

Is it possible to suppress this output?

@Vectorrent
Copy link
Contributor

I don't get that message. But, you could probably suppress it by doing something like:

import logging
logging.getLogger("transformers").setLevel(logging.ERROR)
completion = ai.generate()
logging.getLogger("transformers").setLevel(logging.INFO)

@mspinaci
Copy link

I had the same problem with transformers==4.26.0; just upgrading to 4.27.2 solved it.

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants