Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Dec 7, 2023
1 parent 8468551 commit 07fb3b0
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")


class Detector(DetectorModule):
def __init__(self, model, ds_name):
super().__init__()
Expand Down
1 change: 1 addition & 0 deletions happypose/pose_estimators/megapose/inference/detector.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")


class Detector(DetectorModule):
def __init__(self, model: torch.nn.Module) -> None:
super().__init__()
Expand Down

0 comments on commit 07fb3b0

Please sign in to comment.