AdamLucek commited on
Commit
cbd50d3
1 Parent(s): 3d4fc34

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -50,9 +50,9 @@ Loading in 8-bit Quantization
50
  ```python
51
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
52
 
53
- tokenizer = AutoTokenizer.from_pretrained("AdamLucek/sql-expert")
54
  model = AutoModelForCausalLM.from_pretrained(
55
- "AdamLucek/sql-expert",
56
  device_map="cuda",
57
  quantization_config=BitsAndBytesConfig(load_in_8bit=True)
58
  )
 
50
  ```python
51
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
52
 
53
+ tokenizer = AutoTokenizer.from_pretrained("AdamLucek/llama3-8b-code-sql-slerp")
54
  model = AutoModelForCausalLM.from_pretrained(
55
+ "AdamLucek/llama3-8b-code-sql-slerp",
56
  device_map="cuda",
57
  quantization_config=BitsAndBytesConfig(load_in_8bit=True)
58
  )