add dependence

This commit is contained in:
Guo Liyong 2021-11-06 23:12:48 +08:00
parent a1cdf09655
commit 7b7d710fd1

View File

@ -9,6 +9,7 @@ import subprocess
import sys import sys
from contextlib import contextmanager from contextlib import contextmanager
from pathlib import Path from pathlib import Path
from functools import partial
import torch import torch