Compare commits

...

6 Commits

Author SHA1 Message Date
Dr.Lt.Data
554f767e08
Merge 1670612b90 into d0f3752e33 2025-01-07 21:54:49 -05:00
comfyanonymous
d0f3752e33 Properly calculate inner dim for t5 model.
This is required to support some different types of t5 models.
2025-01-07 17:33:03 -05:00
Dr.Lt.Data
1670612b90 resolve conflict 2024-09-24 21:25:16 +09:00
Dr.Lt.Data
565d67478a
Merge branch 'master' into improve/extra_model_paths_template 2024-09-24 21:22:31 +09:00
Dr.Lt.Data
128075b781
Update extra_model_paths.yaml.example 2024-08-22 13:41:38 +09:00
Dr.Lt.Data
b86567092b add missing folder paths in extra_model_paths template 2024-08-15 10:03:50 +09:00
2 changed files with 7 additions and 1 deletions

View File

@ -227,8 +227,9 @@ class T5(torch.nn.Module):
super().__init__() super().__init__()
self.num_layers = config_dict["num_layers"] self.num_layers = config_dict["num_layers"]
model_dim = config_dict["d_model"] model_dim = config_dict["d_model"]
inner_dim = config_dict["d_kv"] * config_dict["num_heads"]
self.encoder = T5Stack(self.num_layers, model_dim, model_dim, config_dict["d_ff"], config_dict["dense_act_fn"], config_dict["is_gated_act"], config_dict["num_heads"], config_dict["model_type"] != "umt5", dtype, device, operations) self.encoder = T5Stack(self.num_layers, model_dim, inner_dim, config_dict["d_ff"], config_dict["dense_act_fn"], config_dict["is_gated_act"], config_dict["num_heads"], config_dict["model_type"] != "umt5", dtype, device, operations)
self.dtype = dtype self.dtype = dtype
self.shared = operations.Embedding(config_dict["vocab_size"], model_dim, device=device, dtype=dtype) self.shared = operations.Embedding(config_dict["vocab_size"], model_dim, device=device, dtype=dtype)

View File

@ -36,9 +36,14 @@ a111:
# models/diffusion_models # models/diffusion_models
# models/unet # models/unet
# embeddings: models/embeddings/ # embeddings: models/embeddings/
# gligen: models/gligen/
# hypernetworks: models/hypernetworks/
# loras: models/loras/ # loras: models/loras/
# photomaker: models/photomaker/
# style_models: models/style_models/
# upscale_models: models/upscale_models/ # upscale_models: models/upscale_models/
# vae: models/vae/ # vae: models/vae/
# vae_approx: models/vae_approx/
#other_ui: #other_ui:
# base_path: path/to/ui # base_path: path/to/ui