refactai
/

Refact-1_6B-fim

Text Generation

Model card Files Files and versions

svakhreev commited on Aug 29, 2023

Commit

fb08260

·

1 Parent(s): 8188de0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -599,7 +599,7 @@ Fill-in-the-middle uses special tokens to identify the prefix/middle/suffix part
 # pip install -q transformers
 from transformers import AutoModelForCausalLM, AutoTokenizer
-checkpoint = "smallcloudai/Refact-1.6B-fim"
 device = "cuda" # for GPU usage or "cpu" for CPU usage
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)

 # pip install -q transformers
 from transformers import AutoModelForCausalLM, AutoTokenizer
+checkpoint = "smallcloudai/Refact-1_6B-fim"
 device = "cuda" # for GPU usage or "cpu" for CPU usage
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)