Skip to content

这行注释掉会导致微调模型时flashattention2报错(fla2只支持fp16、bf16,关还关不掉),赶紧修复了吧。 #1564

Description

@yy9996

# torch_dtype=torch.float16 if training_args.fp16 else torch.bfloat16,

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions