Update README.md

README.md

@@ -41,17 +41,12 @@ Extensible Configuration: Leverages a custom configuration setup that can be eas
 **How to load and call Kraken-LoRA model :**
 ```
 from transformers import AutoConfig, AutoModelForCausalLM
-from configuration_kraken_lora import KrakenConfig
-from modeling_kraken_lora import KrakenForCausalLM
-
-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 LoRA-expert: