Skip to content

OOM error #34

Open
Open

Description

@Davido111200

Hi,

Thanks for sharing this awesome repo. I'm having memory problem with running run.sh with 1 H100 gpu. I tried reducing batch sizes to 1, but still I can not fine-tune a 7B LLM. Any idea how to work this around?

Activity

  1. Hprairie commented on Dec 18, 2024

    @Hprairie

    Hey did you ever get around this? How much memory was required? I will drop from fp32 to bf16 and reduce the batch size, but wanted to see if I could get any insight. Thanks for the help!

  2. Davido111200 commented on Dec 18, 2024

    @Davido111200
    Author

    I didn't unfortunately. It seems like you gotta try multi-gpu training for this one

  3. Hprairie commented on Dec 18, 2024

    @Hprairie

    Ahh no worries, I appreciate the response!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions