Update README.md
Browse files
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
|
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 |
|