add missing import

used later in line 418
This commit is contained in:
Vladimir Mandic 2023-02-08 07:08:09 -05:00 committed by GitHub
parent ea9bd9fc74
commit 3ee9ca5cb0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,5 +1,6 @@
# this file is adapted from https://github.com/victorca25/iNNfer
from collections import OrderedDict
import math
import functools
import torch