Skip to content

Does not run on CPU only, needs CUDA for compression #6

@wthie

Description

@wthie
./rabbitllm/utils/compression.py", line 192, in compress_layer_state_dict
    v_quant, quant_state = bnb.functional.quantize_nf4(v.cuda(), blocksize=64)
                                                       ^^^^^^^^

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions