(fix): added "model_type" to photomaker node (#6047)

This commit is contained in:
Alexander Piskun 2024-12-15 08:18:02 +03:00 committed by GitHub
parent e83063bf24
commit 1b3a650f19
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -16,6 +16,7 @@ VISION_CONFIG_DICT = {
"patch_size": 14,
"projection_dim": 768,
"hidden_act": "quick_gelu",
"model_type": "clip_vision_model",
}
class MLP(nn.Module):