Found out about air_llm, https://github.com/lyogavin/Anima/tree/main/air_llm, where it loads one layer at a time, allow each layer to be 1.6GB for a 70b with 80 layers. theres about 30mb for kv cache, and i’m not sure where the rest goes.

works with HF out of the box too apparently. The weaknesses appear to be ctxlen, and its gonna be slow, but anyway, anyone want to try goliath 120B unquant?

  • sdmat@alien.topB
    link
    fedilink
    English
    arrow-up
    1
    ·
    1 year ago

    This technique is actually really useful for batch processing.

    I.e. if you run 100 generations and reuse the layer while it is loaded that will go much faster than the total serial time.