From f618e95eba3fc10ac1beabf1d21306986dcc3f52 Mon Sep 17 00:00:00 2001
From: manikbhandari <mbbhandarimanik2@gmail.com>
Date: Wed, 20 Dec 2023 07:55:36 -0500
Subject: [PATCH] remove test gpt2 file

---
 tests/test_gpt2.py | 9 ---------
 1 file changed, 9 deletions(-)
 delete mode 100644 tests/test_gpt2.py

diff --git a/tests/test_gpt2.py b/tests/test_gpt2.py
deleted file mode 100644
index c2cdfed58..000000000
--- a/tests/test_gpt2.py
+++ /dev/null
@@ -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))
\ No newline at end of file