diff --git a/src/models/__init__.py b/src/models/__init__.py index e00c77f..020ae1c 100644 --- a/src/models/__init__.py +++ b/src/models/__init__.py @@ -32,6 +32,7 @@ AVAILABLE_MODELS = { 'HDBLut': hdblut.HDBLut, 'HDBLNet': hdbnet.HDBLNet, 'HDBHNet': hdbnet.HDBHNet, + 'SRMsbLsbR90Net': srnet.SRMsbLsbR90Net, 'SRMsbLsb4R90Net': srnet.SRMsbLsb4R90Net, # 'RCNetCentered_3x3': rcnet.RCNetCentered_3x3, 'RCLutCentered_3x3': rclut.RCLutCentered_3x3, # 'RCNetCentered_7x7': rcnet.RCNetCentered_7x7, 'RCLutCentered_7x7': rclut.RCLutCentered_7x7,