{
  "_name_or_path": "Kelu01/Chem_Fuser",
  "architectures": [
    "HuggingFaceDiffusionModel"
  ],
  "model_type": "gpt2", 
  "auto_map": {
    "AutoConfig": "kelu01/Chem_Fuser--configuration_diffusion.DiffusionConfig",
    "AutoModel": "kelu01/Chem_Fuser--modeling_diffusion.HuggingFaceDiffusionModel"
  },
  "vocab_size": 425,
  "d_model": 512,
  "n_heads": 16,
  "n_layers": 6,
  "dropout": 0.1,
  "max_len": 2250,
  "num_steps": 1000,
  "pad_token": 0,
  "torch_dtype": "float32",
  "transformers_version": "4.46.3",
  "custom_pipelines": {
    "smiles-diffusion": {
      "impl": "pipe.SmilesDiffusionPipe",
      "pt": [
        "AutoModel"
      ],
      "tf": [],
      "type": "text"
    }
  }
}