updates

5 months ago · 8257ec7e57
parent 90c9f2869e
commit 8257ec7e57
1 changed files with 1 additions and 0 deletions
--- 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,