MogensR commited on
Commit
3e4b062
·
verified ·
1 Parent(s): 508dfbb

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 = 4 # Support both RGB and RGB+prob inputs
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)