shainaraza commited on
Commit
32bed4f
1 Parent(s): 50e4378

Update my_toxicity_debiaser.py

Browse files
Files changed (1) hide show
  1. 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