Skip to content

Commit 2d79a25

Browse files
register presets (#2268)
1 parent d4937be commit 2d79a25

File tree

3 files changed

+14
-0
lines changed

3 files changed

+14
-0
lines changed
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
from keras_hub.src.models.mixtral.mixtral_backbone import MixtralBackbone
2+
from keras_hub.src.models.mixtral.mixtral_presets import backbone_presets
3+
from keras_hub.src.utils.preset_utils import register_presets
4+
5+
register_presets(backbone_presets, MixtralBackbone)
Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,5 @@
11
from keras_hub.src.models.qwen.qwen_backbone import QwenBackbone
2+
from keras_hub.src.models.qwen.qwen_presets import backbone_presets
3+
from keras_hub.src.utils.preset_utils import register_presets
4+
5+
register_presets(backbone_presets, QwenBackbone)
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
from keras_hub.src.models.qwen_moe.qwen_moe_backbone import QwenMoeBackbone
2+
from keras_hub.src.models.qwen_moe.qwen_moe_presets import backbone_presets
3+
from keras_hub.src.utils.preset_utils import register_presets
4+
5+
register_presets(backbone_presets, QwenMoeBackbone)

0 commit comments

Comments
 (0)