mwitiderrick commited on
Commit
2777e2e
1 Parent(s): 1d15734

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -68,8 +68,8 @@ This is an [OpenLlama model Code Instruct](https://huggingface.co/mwitiderrick/o
68
  ```python
69
  from transformers import AutoTokenizer, AutoModelForCausalLM,pipeline
70
 
71
- tokenizer = AutoTokenizer.from_pretrained("mwitiderrick/open_llama_3b_glaive_assistant_v0.1")
72
- model = AutoModelForCausalLM.from_pretrained("mwitiderrick/open_llama_3b_glaive_assistant_v0.1")
73
  query = "Write a quick sort algorithm in Python"
74
  text_gen = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
75
  output = text_gen(f"<s>[INST]{query}[/INST]")
 
68
  ```python
69
  from transformers import AutoTokenizer, AutoModelForCausalLM,pipeline
70
 
71
+ tokenizer = AutoTokenizer.from_pretrained("mwitiderrick/open_llama_3b_glaive_v0.1")
72
+ model = AutoModelForCausalLM.from_pretrained("mwitiderrick/open_llama_3b_glaive_v0.1")
73
  query = "Write a quick sort algorithm in Python"
74
  text_gen = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
75
  output = text_gen(f"<s>[INST]{query}[/INST]")