Chemfuser / config.json
kelu01's picture
Update config.json
c350f32 verified
Raw History Blame Contribute Delete
706 Bytes
{
"_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"
}
}
}