From 8257ec7e579368529d703aa69619d95913a883db Mon Sep 17 00:00:00 2001 From: protsenkovi Date: Tue, 11 Jun 2024 12:19:38 +0400 Subject: [PATCH] updates --- src/models/__init__.py | 1 + 1 file changed, 1 insertion(+) 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,