Skip to content
New issue

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

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

Already on GitHub? Sign in to your account

Upgrade jax/jaxlib/flax pin versions #24791

Merged
merged 1 commit into from
Jul 13, 2023
Merged

Upgrade jax/jaxlib/flax pin versions #24791

merged 1 commit into from
Jul 13, 2023

Conversation

ydshieh
Copy link
Collaborator

@ydshieh ydshieh commented Jul 12, 2023

What does this PR do?

So we can have latest TF versions.

@ydshieh ydshieh requested a review from sanchit-gandhi July 12, 2023 21:17
@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint.

Copy link
Contributor

@sanchit-gandhi sanchit-gandhi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM - thanks @ydshieh. For context, these are also valid versions of JAX/JAX lib / Flax in diffusers (see setup.py#L94, so we can be fairly sure they are compatible with our CI given the dependency of diffusers on transformers Also cc @cgarciae for interest

@sanchit-gandhi sanchit-gandhi linked an issue Jul 13, 2023 that may be closed by this pull request
@ydshieh ydshieh merged commit e538189 into main Jul 13, 2023
Lorenzobattistela pushed a commit to Lorenzobattistela/transformers that referenced this pull request Jul 13, 2023
@ydshieh ydshieh deleted the check_flax branch August 29, 2023 15:33
blbadger pushed a commit to blbadger/transformers that referenced this pull request Nov 8, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Un-pin JAX from <= 0.3.6
3 participants