Chemfuser / config.json
kelu01's picture
Update config.json
39f773e verified
Raw History Blame
686 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": 103,
"d_model": 256,
"n_heads": 8,
"n_layers": 6,
"dropout": 0.1,
"max_len": 140,
"num_steps": 1000,
"torch_dtype": "float32",
"transformers_version": "4.46.3",
"custom_pipelines": {
"smiles-diffusion": {
"impl": "pipe.SmilesDiffusionPipe",
"pt": [
"AutoModel"
],
"tf": [],
"type": "text"
}
}
}