vishal042002
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -60,7 +60,6 @@ LOADING THE MODEL DIRCTLY:
|
|
60 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
61 |
import torch
|
62 |
|
63 |
-
# Use the merged model name from your Hugging Face repository
|
64 |
model_name = "vishal042002/Llama3.2-3b-Instruct-ClinicalSurgery"
|
65 |
model = AutoModelForCausalLM.from_pretrained(
|
66 |
model_name,
|
@@ -68,7 +67,6 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
68 |
device_map="auto"
|
69 |
)
|
70 |
|
71 |
-
# Load the tokenizer
|
72 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
73 |
```
|
74 |
|
|
|
60 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
61 |
import torch
|
62 |
|
|
|
63 |
model_name = "vishal042002/Llama3.2-3b-Instruct-ClinicalSurgery"
|
64 |
model = AutoModelForCausalLM.from_pretrained(
|
65 |
model_name,
|
|
|
67 |
device_map="auto"
|
68 |
)
|
69 |
|
|
|
70 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
71 |
```
|
72 |
|