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 |
|