mrm8488
/

bloom-560m-finetuned-the-stack-rust

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

mrm8488 commited on Dec 30, 2022

Commit

5358462

·

1 Parent(s): 88fe318

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ from transformers import BloomTokenizerFast, BloomForCausalLM
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 ckpt = 'mrm8488/bloom-560m-finetuned-the-stack-rust'
-revision = '100k' #latest one at the moment
 tokenizer = BloomTokenizerFast.from_pretrained(ckpt)
 model = BloomForCausalLM.from_pretrained(ckpt, revision=revision).to(device)

 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 ckpt = 'mrm8488/bloom-560m-finetuned-the-stack-rust'
+revision = '100k' # latest one at the moment
 tokenizer = BloomTokenizerFast.from_pretrained(ckpt)
 model = BloomForCausalLM.from_pretrained(ckpt, revision=revision).to(device)