vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

remove test gpt2 file

Browse source
This commit is contained in:
manikbhandari 2023-12-20 07:55:36 -05:00
parent 92b6fc3db2
commit f618e95eba
1 changed files with 0 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/test_gpt2.py
View file

@ -1,9 +0,0 @@
from transformers import set_seed
from transformers import AutoModelForCausalLM, pipeline, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("gpt2", cache_dir="models")
tokenizer = AutoTokenizer.from_pretrained("gpt2", cache_dir="models")
generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
set_seed(42)
print(generator("Hello, I'm a language model,", max_length=30, num_return_sequences=5))