Skip to content

Add support GPU in the model #30

Open
Open
Add support GPU in the model#30
@SarahAlidoost

Description

@SarahAlidoost
SarahAlidoost
opened on Mar 2, 2026

Implement device and dtype in the model to support GPU and dtypes. Related to #21

  • don't use torch.set_default_*
  • add a config for setting computation resources
  • skip test as github runner doesn't have gpu

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