Anonymous commited on
Commit
3c83bdd
·
1 Parent(s): d344035

remove transformers

Browse files
Files changed (1) hide show
  1. tasks/qa.py +0 -4
tasks/qa.py CHANGED
@@ -42,10 +42,6 @@ def gemini_completion(prompt):
42
  # model.to("cuda:04")
43
 
44
 
45
-
46
- from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
47
-
48
-
49
  def get_entities_gpt3_long(prompt):
50
  response = openai.ChatCompletion.create(
51
  engine="chatgpt", temperature=0, messages=[{"role": "user", "content": prompt}]
 
42
  # model.to("cuda:04")
43
 
44
 
 
 
 
 
45
  def get_entities_gpt3_long(prompt):
46
  response = openai.ChatCompletion.create(
47
  engine="chatgpt", temperature=0, messages=[{"role": "user", "content": prompt}]