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

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