Skip to content

Conversation

@orbitalquark
Copy link
Contributor

#461 increased memory usage because streaming is not supported when running in ICL mode.

This PR enables streaming for ICL mode, drastically reducing memory usage again. Some quick eyeballing shows a decrease in peak memory usage from ~10GB to ~6GB.

This code was adapted from the streaming code in generate().

@orbitalquark
Copy link
Contributor Author

The test failure appears to be in Dia and seems unrelated to this PR. I don't know how to re-run the test to see if it is a sporadic one.

Copy link
Collaborator

@lucasnewman lucasnewman left a comment

Choose a reason for hiding this comment

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

Thanks, looks good!The tests were pulling network data unauthenticated to HF which was triggering timeouts -- it should be more reliable now.

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.

2 participants