23 lines
926 B
Python
23 lines
926 B
Python
|
"""
|
||
|
|
Adapted from https://github.com/XPixelGroup/BasicSR
|
||
|
|
License: Apache-2.0
|
||
|
|
|
||
|
|
As of Feb 2024, `basicsr` appears to be unmaintained. It imports a function from `torchvision` that is removed in
|
||
|
|
`torchvision` 0.17. Here is the deprecation warning:
|
||
|
|
|
||
|
|
UserWarning: The torchvision.transforms.functional_tensor module is deprecated in 0.15 and will be **removed in
|
||
|
|
0.17**. Please don't rely on it. You probably just need to use APIs in torchvision.transforms.functional or in
|
||
|
|
torchvision.transforms.v2.functional.
|
||
|
|
|
||
|
|
As a result, a dependency on `basicsr` means we cannot keep our `torchvision` dependency up to date.
|
||
|
|
|
||
|
|
Because we only rely on a single class `RRDBNet` from `basicsr`, we've copied the relevant code here and removed the
|
||
|
|
dependency on `basicsr`.
|
||
|
|
|
||
|
|
The code is almost unchanged, only a few type annotations have been added. The license is also copied.
|
||
|
|
|
||
|
|
Copy From InvokeAI
|
||
|
|
"""
|
||
|
|
|
||
|
|
from .rrdbnet_arch import RRDBNet
|