CognitivessAI
/

cognitivess

Text Generation

text-generation-inference

Sentiment Analysis

Inference Endpoints

8-bit precision

Model card Files Files and versions Community

cognitivess commited on Jul 20

Commit

6e88f4c

•

1 Parent(s): 4defe05

Update cognitivess_model/init.py

Files changed (1) hide show

cognitivess_model/__init__.py +7 -4

cognitivess_model/__init__.py CHANGED Viewed

@@ -5,13 +5,16 @@ from .tokenization_cognitivess_fast import CognitivessTokenizerFast
 from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer
-# Register the custom config
 AutoConfig.register("cognitivess", CognitivessConfig)
-# Register the custom model
 AutoModelForCausalLM.register(CognitivessConfig, CognitivessForCausalLM)
-# Register the custom tokenizer
 AutoTokenizer.register(CognitivessConfig,
                        slow_tokenizer_class=CognitivessTokenizer,
-                       fast_tokenizer_class=CognitivessTokenizerFast)

 from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer
+print("Registering CognitivessConfig...")
 AutoConfig.register("cognitivess", CognitivessConfig)
+print("CognitivessConfig registered successfully.")
+print("Registering CognitivessForCausalLM...")
 AutoModelForCausalLM.register(CognitivessConfig, CognitivessForCausalLM)
+print("CognitivessForCausalLM registered successfully.")
+print("Registering CognitivessTokenizer and CognitivessTokenizerFast...")
 AutoTokenizer.register(CognitivessConfig,
                        slow_tokenizer_class=CognitivessTokenizer,
+                       fast_tokenizer_class=CognitivessTokenizerFast)
+print("CognitivessTokenizer and CognitivessTokenizerFast registered successfully.")