aboutsummaryrefslogtreecommitdiff
path: root/modules/esrgan_model_arch.py
diff options
context:
space:
mode:
Diffstat (limited to 'modules/esrgan_model_arch.py')
-rw-r--r--modules/esrgan_model_arch.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/esrgan_model_arch.py b/modules/esrgan_model_arch.py
index bc9ceb2a..1b52b0f5 100644
--- a/modules/esrgan_model_arch.py
+++ b/modules/esrgan_model_arch.py
@@ -1,5 +1,6 @@
# this file is adapted from https://github.com/victorca25/iNNfer
+from collections import OrderedDict
import math
import functools
import torch