AI-Hypercomputer / jetstream-pytorch

PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
Apache License 2.0
41 stars 15 forks source link

fix: correct error message #193

Open tengomucho opened 1 month ago

tengomucho commented 1 month ago

It was supposed to print the received array but it was not doing it. Also, the type is now displayed.

cc: @qihqi