Update README.md

README.md

@@ -8,7 +8,7 @@ tags:
 - solar
 license: apache-2.0
 ---
-# Nous-Hermes-2-SOLAR-18B
+# vicgalle/franken-SOLAR-18B-v1.0
 
 This is a SOLAR-like model upscaled to 18B. 
 It is a frankenmerge model created using mergekit, alternating layers of Nous-Hermes-2-SOLAR-10.7B and SOLAR-10.7B-Instruct.
@@ -69,8 +69,8 @@ dtype: float16
 You can use the provided template:
 
 ```
-tokenizer = AutoTokenizer.from_pretrained("vicgalle/Nous-Hermes-2-SOLAR-18B")
-model = AutoModelForCausalLM.from_pretrained("vicgalle/Nous-Hermes-2-SOLAR-18B", torch_dtype=torch.float16, load_in_4bit=True)
+tokenizer = AutoTokenizer.from_pretrained("vicgalle/franken-SOLAR-18B-v1.0")
+model = AutoModelForCausalLM.from_pretrained("vicgalle/franken-SOLAR-18B-v1.0", torch_dtype=torch.float16, load_in_4bit=True)
 
 conversation = [ {'role': 'system', 'content': SYSTEM_PROMPT}, {'role': 'user', 'content': USER_PROMPT} ] 
 prompt = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)