[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
5df56c0804
commit
310849726e
1 changed files with 1 additions and 0 deletions
|
|
@ -236,6 +236,7 @@ def _patch_capability(monkeypatch, capability):
|
|||
import torch
|
||||
|
||||
import core.training.diffusion_train_common as dtc
|
||||
|
||||
monkeypatch.setattr(torch.cuda, "is_available", lambda: True)
|
||||
monkeypatch.setattr(torch.cuda, "get_device_capability", lambda *a, **k: capability)
|
||||
monkeypatch.setattr(dtc, "has_functional_torchao", lambda: True)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue