Skip to content

Conversation

@qihqi
Copy link
Collaborator

@qihqi qihqi commented Jun 11, 2024

No description provided.

@qihqi qihqi requested review from FanhaiLu1 and wang2yn84 June 11, 2024 21:02
Copy link
Collaborator

@FanhaiLu1 FanhaiLu1 left a comment

Choose a reason for hiding this comment

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

Hi Han, did you test the script on GPU vms?

@wang2yn84
Copy link
Collaborator

Seems the only difference is jax[tpu] vs jax[cuda]?

@qihqi qihqi merged commit e2ee7dd into main Jun 12, 2024
@qihqi qihqi deleted the hanq_fix2 branch June 12, 2024 00:11
@qihqi
Copy link
Collaborator Author

qihqi commented Jun 12, 2024

Hi Han, did you test the script on GPU vms?

yes both run_interactive and run_server ran; but perf is bad

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.

4 participants