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

Save cached latents as caching progresses #39

Open
wants to merge 4 commits into
base: main
Choose a base branch
from

Conversation

kernel-loophole
Copy link

@kernel-loophole kernel-loophole commented Nov 18, 2024

  • Added support for saving latents after processing a configurable number of tokens (save_every) to reduce memory usage.
  • Introduced a temp_dir in the Cache class for managing intermediate cache files.
  • Implemented a merge_temp_files method to combine temporary files into a final cache.
    @SrGonao make changes .i am facing some issue set the temp_dir .will be thankful to you if you review the changes.
    Save cached latents as caching progresses #38

@CLAassistant
Copy link

CLAassistant commented Nov 18, 2024

CLA assistant check
All committers have signed the CLA.

@SrGonao
Copy link
Collaborator

SrGonao commented Nov 25, 2024

I don't see the merge tempfiles method

@kernel-loophole
Copy link
Author

hi there ,i write the merge_temp_files ,can you review it .

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.

3 participants