blapuma commited on
Commit
fcc5a19
·
verified ·
1 Parent(s): 9e9135c

Upload MyTestPipeline

Browse files
Files changed (1) hide show
  1. new_task.py +0 -6
new_task.py CHANGED
@@ -4,12 +4,6 @@ import tensorflow as tf
4
  import numpy as np
5
 
6
  class MyTestPipeline(TextGenerationPipeline):
7
- def __init__(self, *args, **kwargs):
8
- kwargs['trust_remote_code'] = True
9
- self.trust_remote_code = True
10
- super().__init__(*args, **kwargs)
11
- self.trust_remote_code = True
12
-
13
  def preprocess(self, text, **kwargs):
14
  prompt = 'Answer the following question/statement without any explanation, do not abbreviate names, and give just the answer.'
15
  txt = f"<|user|>\n{prompt} {text}\n<|end|>\n<|assistant|>"
 
4
  import numpy as np
5
 
6
  class MyTestPipeline(TextGenerationPipeline):
 
 
 
 
 
 
7
  def preprocess(self, text, **kwargs):
8
  prompt = 'Answer the following question/statement without any explanation, do not abbreviate names, and give just the answer.'
9
  txt = f"<|user|>\n{prompt} {text}\n<|end|>\n<|assistant|>"