shainaraza
commited on
Commit
•
32bed4f
1
Parent(s):
50e4378
Update my_toxicity_debiaser.py
Browse files- my_toxicity_debiaser.py +0 -9
my_toxicity_debiaser.py
CHANGED
@@ -55,14 +55,5 @@ class MyToxicityDebiaserPipeline(object):
|
|
55 |
outputs = self._forward(inputs, **forward_kwargs)
|
56 |
return self.postprocess(outputs)
|
57 |
|
58 |
-
# Create an instance of the custom pipeline
|
59 |
-
custom_pipeline = MyToxicityDebiaserPipeline(
|
60 |
-
model=toxicity_model,
|
61 |
-
tokenizer=toxicity_tokenizer,
|
62 |
-
gpt_model=gpt_model,
|
63 |
-
gpt_tokenizer=gpt_tokenizer,
|
64 |
-
device=torch.device("cuda" if torch.cuda.is_available() else "cpu"),
|
65 |
-
binary_output=True
|
66 |
-
)
|
67 |
|
68 |
|
|
|
55 |
outputs = self._forward(inputs, **forward_kwargs)
|
56 |
return self.postprocess(outputs)
|
57 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
58 |
|
59 |
|