Update README.md

README.md

@@ -31,18 +31,13 @@ Extensible Configuration: Leverages a custom configuration setup that can be eas
 
 **How to load and call Kraken model :**
 ```
-from transformers import AutoConfig, AutoModelForCausalLM
-from configuration_kraken import KrakenConfig
-from modeling_kraken import KrakenForCausalLM
+from transformers import AutoModelForCausalLM
 
-AutoConfig.register("kraken", KrakenConfig)
-AutoModelForCausalLM.register(KrakenConfig, KrakenForCausalLM)
 
 device = "cuda:0" ## Setup "cuda:0" if NVIDIA, "mps" if on Mac
 
 # Load the model and config:
-config = AutoConfig.from_pretrained("./kraken_model")
-model = AutoModelForCausalLM.from_pretrained("./kraken_model", config=config, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("./kraken_model", trust_remote_code=True)
 ```
 
 # Call the Reasoning expert: