Spaces:
Running
Running
InterFuser Model Directory
Required Files
Place your trained InterFuser model files in this directory:
interfuser_model.pth
- The main model weights fileconfig.json
(optional) - Model configuration file
Model Format
The model should be a PyTorch state dict saved with:
torch.save(model.state_dict(), 'interfuser_model.pth')
Loading in Code
The model is loaded in model_definition.py
:
model = InterFuserModel()
model.load_state_dict(torch.load('model/interfuser_model.pth', map_location='cpu'))
Note
- The current implementation uses a dummy model for testing
- Replace with your actual trained InterFuser weights
- Ensure the model architecture matches the one defined in
model_definition.py