HyperStyler / config.json
JKSHINPG's picture
Release HyperStyler (Reddit)
834e995 verified
Raw History Blame Contribute Delete
883 Bytes
{
"architectures": [
"HyperStylerModel"
],
"model_type": "hyperstyler",
"auto_map": {
"AutoConfig": "configuration_hyperstyler.HyperStylerConfig",
"AutoModel": "modeling_hyperstyler.HyperStylerModel"
},
"base_model_name": "google/t5-v1_1-large",
"max_encoder_len": 80,
"max_decoder_len": 80,
"styler_config": {
"ctrl_dim": 768,
"use_hypernet": true,
"hypernet_config": {
"target_layers": [
"crossattn",
"ffn"
],
"num_prefix": 5,
"rank": 32,
"parameter_efficiency_mode": false,
"bias": false,
"search_config": "MHA",
"dropout": 0.1
},
"use_stylonavigator": true,
"stylo_navigator_config": {
"dropout": 0.1,
"temperature": 1.0
}
},
"style_model_name": "AnnaWegmann/Style-Embedding",
"torch_dtype": "float32",
"transformers_version": "4.50.0"
}