Skip to content

Remove pjrt config JAX_USE_PJRT_C_API_ON_TPU from decode.py#27

Closed
gobbleturk wants to merge 1 commit into
mainfrom
chatbot
Closed

Remove pjrt config JAX_USE_PJRT_C_API_ON_TPU from decode.py#27
gobbleturk wants to merge 1 commit into
mainfrom
chatbot

Conversation

@gobbleturk
Copy link
Copy Markdown
Collaborator

We removed the pjrt config option since it is set to 1 by default now.

@gobbleturk gobbleturk requested a review from rwitten May 4, 2023 20:19
@gobbleturk gobbleturk changed the title Update decode.py to remove pjrt flag Update decode.py to remove pjrt config May 4, 2023
@gobbleturk gobbleturk changed the title Update decode.py to remove pjrt config Remove pjrt config (JAX_USE_PJRT_C_API_ON_TPU) from decode.py May 4, 2023
@gobbleturk gobbleturk changed the title Remove pjrt config (JAX_USE_PJRT_C_API_ON_TPU) from decode.py Remove pjrt config JAX_USE_PJRT_C_API_ON_TPU from decode.py May 4, 2023
@gobbleturk gobbleturk closed this Jun 9, 2023
phu0ngng added a commit to phu0ngng/maxtext that referenced this pull request Oct 22, 2025
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.

1 participant