Fix Large model.

app.py

@@ -198,13 +198,9 @@ class REncoderLarge(nn.Module):
 
 class RDecoderLarge(nn.Module):
 
-    def __init__(self, args):
+    def __init__(self, args, dd, ee, ff):
         super().__init__()
         self.args = args
-        if 'd16_512' in model_prefix:
-            dd, ee, ff = 16, 64, 512
-        elif 'd32_1024' in model_prefix:
-            dd, ee, ff = 32, 128, 1024
         self.CZZ = nn.Conv2d(args.my_img_bit,
                              dd * 64,
                              kernel_size=3,