matthewlyleolson commited on
Commit
e14d881
1 Parent(s): 3371fdb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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-old"
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)