Spaces:
Runtime error
Runtime error
Fix Large model.
Browse files
app.py
CHANGED
@@ -198,13 +198,9 @@ class REncoderLarge(nn.Module):
|
|
198 |
|
199 |
class RDecoderLarge(nn.Module):
|
200 |
|
201 |
-
def __init__(self, args):
|
202 |
super().__init__()
|
203 |
self.args = args
|
204 |
-
if 'd16_512' in model_prefix:
|
205 |
-
dd, ee, ff = 16, 64, 512
|
206 |
-
elif 'd32_1024' in model_prefix:
|
207 |
-
dd, ee, ff = 32, 128, 1024
|
208 |
self.CZZ = nn.Conv2d(args.my_img_bit,
|
209 |
dd * 64,
|
210 |
kernel_size=3,
|
|
|
198 |
|
199 |
class RDecoderLarge(nn.Module):
|
200 |
|
201 |
+
def __init__(self, args, dd, ee, ff):
|
202 |
super().__init__()
|
203 |
self.args = args
|
|
|
|
|
|
|
|
|
204 |
self.CZZ = nn.Conv2d(args.my_img_bit,
|
205 |
dd * 64,
|
206 |
kernel_size=3,
|