DavidGF commited on
Commit
ee5d71a
1 Parent(s): 2e9a091

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -40,12 +40,11 @@ Extensible Configuration: Leverages a custom configuration setup that can be eas
40
 
41
  **How to load and call Kraken-LoRA model :**
42
  ```
43
- from transformers import AutoConfig, AutoModelForCausalLM
44
 
45
  device = "cuda:0" ## Setup "cuda:0" if NVIDIA, "mps" if on Mac
46
 
47
  # Load the model and config:
48
- config = AutoConfig.from_pretrained("./kraken_model")
49
  model = AutoModelForCausalLM.from_pretrained("./kraken_model", trust_remote_code=True)
50
  ```
51
 
 
40
 
41
  **How to load and call Kraken-LoRA model :**
42
  ```
43
+ from transformers import AutoModelForCausalLM
44
 
45
  device = "cuda:0" ## Setup "cuda:0" if NVIDIA, "mps" if on Mac
46
 
47
  # Load the model and config:
 
48
  model = AutoModelForCausalLM.from_pretrained("./kraken_model", trust_remote_code=True)
49
  ```
50