Update README.md

README.md

@@ -39,6 +39,8 @@ from peft import LoraConfig, get_peft_model
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 
+login() # Need access to the gated model.
+
 # Load LLAMA 2 model
 model_name = "meta-llama/Llama-2-13b-chat-hf" 
 
@@ -78,17 +80,3 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 This loads the LLAMA 2 model, applies 4-bit quantization and LoRA optimizations, constructs a prompt, and generates a response.
 
 See the [docs](https://huggingface.co/docs/transformers/model_doc/auto#transformers.AutoModelForCausalLM) for more details.
-
-## Training
-
-The model was trained by Anthropic using self-supervised learning. See the [model card](https://huggingface.co/USERNAME/archimedes) for details.
-
-## License
-
-Archimedes is released under the Apache 2.0 license.
-
-## Citation
-
-Coming soon!
-
-Please ⭐ if this repository was helpful!