QM9BaseModel

class QM9BaseModel(device: device)[source]

Bases: FlowMolBaseModel

Pre-trained continuous flow matching model, trained on QM9.