Upload SegmentBorzoi

segment_borzoi.py

@@ -103,6 +103,11 @@ class SegmentBorzoi(PreTrainedModel):
                 num_rel_pos_features=config.num_rel_pos_features,
             )
 
+        # Correct conv layer in downsample block
+        self.unet_head.unet.downsample_blocks[0].conv_layers[0] = nn.Conv1d(
+            in_channels=1920, out_channels=1536, kernel_size=3, stride=1, padding=1
+        )
+
         # Correct bias in separable layers
         self.separable1.conv_layer[1].bias = None
         self.separable0.conv_layer[1].bias = None