matthewlyleolson
commited on
Commit
•
e14d881
1
Parent(s):
3371fdb
Update README.md
Browse files
README.md
CHANGED
@@ -96,7 +96,7 @@ def add_model_a_to_b(model_a, model_b):
|
|
96 |
model_b.load_state_dict(state_dict_b)
|
97 |
|
98 |
output_checkpoint = "" # set if you don't want to merge every time
|
99 |
-
hf_checkpoint = "Intel/llava-llama-3-8b
|
100 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
101 |
|
102 |
processor = AutoProcessor.from_pretrained(hf_checkpoint)
|
|
|
96 |
model_b.load_state_dict(state_dict_b)
|
97 |
|
98 |
output_checkpoint = "" # set if you don't want to merge every time
|
99 |
+
hf_checkpoint = "Intel/llava-llama-3-8b"
|
100 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
101 |
|
102 |
processor = AutoProcessor.from_pretrained(hf_checkpoint)
|