Skip to content

failed to inference on pure cpu device #22

@ZoengMingWong

Description

@ZoengMingWong

The project seems not support pure cpu inference, i.e. run example.py with --device cpu, as the attention modules always use xformers with attn_bias is not None.

ValueError: Attention bias and Query/Key/Value should be on the same device
query.device: cpu
attn_bias : cuda:0

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    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