GEOMBaseModel

class GEOMBaseModel(device: device)[source]

Bases: FlowMolBaseModel

Pre-trained continuous flow matching model, trained on GEOM-Drugs.