speechisalluneed commited on
Commit
3cbab94
·
verified ·
1 Parent(s): f165406

Update __init__.py

Browse files
Files changed (1) hide show
  1. __init__.py +6 -5
__init__.py CHANGED
@@ -1,7 +1,8 @@
1
- from .configuration_fastslm import FastSLMConfig
2
- from .modeling_fastslm import FastSLMForConditionalGeneration # FastALMForCausalLM
3
 
4
- from transformers import AutoConfig, AutoModelForCausalLM
5
 
6
- AutoConfig.register("fastslm", FastSLMConfig)
7
- AutoModelForCausalLM.register(FastSLMConfig, FastSLMForConditionalGeneration)
 
 
1
+ from .configuration_symphony import SymphonyConfig
2
+ from .modeling_symphony import SymphonyForConditionalGeneration
3
 
4
+ from transformers import AutoConfig, AutoModelForCausalLM, AutoModel
5
 
6
+ AutoConfig.register("symphony", SymphonyConfig)
7
+ AutoModelForCausalLM.register(SymphonyConfig, SymphonyForConditionalGeneration)
8
+ AutoModel.register(SymphonyConfig, SymphonyForConditionalGeneration)