Update matanyone_fixed/utils/get_default_model.py
Browse files
matanyone_fixed/utils/get_default_model.py
CHANGED
|
@@ -186,7 +186,7 @@ def get_matanyone_model(checkpoint_path: Union[str, Path],
|
|
| 186 |
# Determine input channels based on usage
|
| 187 |
# If we're using probability guidance, we need 4 channels (RGB + prob)
|
| 188 |
# Otherwise, 3 channels (RGB only)
|
| 189 |
-
input_channels =
|
| 190 |
|
| 191 |
# Create model
|
| 192 |
model = SimpleMatteModel(backbone_channels=input_channels)
|
|
|
|
| 186 |
# Determine input channels based on usage
|
| 187 |
# If we're using probability guidance, we need 4 channels (RGB + prob)
|
| 188 |
# Otherwise, 3 channels (RGB only)
|
| 189 |
+
input_channels = 3 # Support both RGB and RGB+prob inputs
|
| 190 |
|
| 191 |
# Create model
|
| 192 |
model = SimpleMatteModel(backbone_channels=input_channels)
|