diff options
Diffstat (limited to 'modules/codeformer')
| -rw-r--r-- | modules/codeformer/codeformer_arch.py | 2 | ||||
| -rw-r--r-- | modules/codeformer/vqgan_arch.py | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/modules/codeformer/codeformer_arch.py b/modules/codeformer/codeformer_arch.py index 0eff93dc..11dcc3ee 100644 --- a/modules/codeformer/codeformer_arch.py +++ b/modules/codeformer/codeformer_arch.py @@ -1,3 +1,5 @@ +# this file is copied from CodeFormer repository. Please see comment in modules/codeformer_model.py + import math import numpy as np import torch diff --git a/modules/codeformer/vqgan_arch.py b/modules/codeformer/vqgan_arch.py index f6dfcf4c..c06c590c 100644 --- a/modules/codeformer/vqgan_arch.py +++ b/modules/codeformer/vqgan_arch.py @@ -1,3 +1,5 @@ +# this file is copied from CodeFormer repository. Please see comment in modules/codeformer_model.py + ''' VQGAN code, adapted from the original created by the Unleashing Transformers authors: https://github.com/samb-t/unleashing-transformers/blob/master/models/vqgan.py |