Update README.md

README.md

@@ -12,7 +12,8 @@ tags:
 # Strix Rufipes 70B
 
 # Model Details
-* **Trained by**: ibivibiv
+* **Trained by**: [ibivibiv](https://huggingface.co/ibivibiv)
+* **Library**: [HuggingFace Transformers](https://github.com/huggingface/transformers)
 * **Model type:**  **strix-rufipes-70b** is an auto-regressive language model fine tuned on the Llama 2 transformer architecture.
 * **Language(s)**: English
 * **Purpose**: Has specific training for logic enforcement, will do well in ARC or other logic testing as well as critical thinking tasks.  This model is targeted towards planning exercises.
@@ -40,7 +41,7 @@ torch.set_default_device("cuda")
 model = AutoModelForCausalLM.from_pretrained("ibivibiv/strix-rufipes-70b", torch_dtype="auto", device_config='auto')
 tokenizer = AutoTokenizer.from_pretrained("ibivibiv/strix-rufipes-70b", trust_remote_code=True)
 
-inputs = tokenizer("Create a plan for developing the game of snake in python using pygame.", return_tensors="pt", return_attention_mask=False)
+inputs = tokenizer("### Instruction: Create a plan for developing the game of snake in python using pygame.\n### Response:\n", return_tensors="pt", return_attention_mask=False)
 
 outputs = model.generate(**inputs, max_length=200)
 text = tokenizer.batch_decode(outputs)[0]