Skip to content
This repository has been archived by the owner on Jun 24, 2024. It is now read-only.

Add scratchbuffers to MPT #278

Merged
merged 1 commit into from
May 25, 2023
Merged

Add scratchbuffers to MPT #278

merged 1 commit into from
May 25, 2023

Conversation

LLukas22
Copy link
Contributor

Add scratchbuffers similar to the upstream ggml examples of GptNeoX/MPT. Should reduce memory consumption.

@@ -46,15 +46,15 @@ fn will_fail_on_scored_ggml_save() {
}

#[test]
fn can_roundtrip_loader_and_saver_ggjt_v2() {
fn can_roundtrip_loader_and_saver_ggjt_v3() {
Copy link
Collaborator

Choose a reason for hiding this comment

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

oops good catch

@philpax philpax merged commit 5cb475a into rustformers:main May 25, 2023
@LLukas22 LLukas22 deleted the feat/mpt-scratchbuffer branch June 12, 2023 13:07
@hhamud hhamud mentioned this pull request Aug 7, 2023
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants