Update README.md

README.md

@@ -9,7 +9,7 @@ language:
 - en
 pipeline_tag: text-generation
 ---
-# FreeWilly
+# StableBeluga_13B
 
 ## Model Description
 
@@ -25,7 +25,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
 tokenizer = AutoTokenizer.from_pretrained("stabilityai/StableBeluga_13B", use_fast=False)
 model = AutoModelForCausalLM.from_pretrained("stabilityai/StableBeluga_13B", torch_dtype=torch.float16, low_cpu_mem_usage=True, device_map="auto")
-system_prompt = "### System:\nYou are Free Willy, an AI that follows instructions extremely well. Help as much as you can. Remember, be safe, and don't do anything illegal.\n\n"
+system_prompt = "### System:\nYou are StableBeluga_13B, an AI that follows instructions extremely well. Help as much as you can. Remember, be safe, and don't do anything illegal.\n\n"
 
 message = "Write me a poem please"
 prompt = f"{system_prompt}### User: {message}\n\n### Assistant:\n"
@@ -35,7 +35,7 @@ output = model.generate(**inputs, do_sample=True, top_p=0.95, top_k=0, max_new_t
 print(tokenizer.decode(output[0], skip_special_tokens=True))
 ```
 
-FreeWilly should be used with this prompt format:
+StableBeluga_13B should be used with this prompt format:
 ```
 ### System:
 This is a system prompt, please behave and help the user.
@@ -44,7 +44,7 @@ This is a system prompt, please behave and help the user.
 Your prompt here
 
 ### Assistant
-The output of FreeWilly2
+The output of StableBeluga_13B
 ```
 
 ## Model Details