Skip to content

GitLab

Unverified Commit 999540df authored by Nicolas Patry's avatar Nicolas Patry Committed by GitHub
Browse files

Tensor location is already handled (#14224) 

in `base.py` not in subclasses.
parent 323f28dc
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 4 deletions
src/transformers/pipelines/text_classification.py
View file @ 999540df
......@@ -145,9 +145,6 @@ class TextClassificationPipeline(Pipeline):
function_to_apply = ClassificationFunction.NONE
outputs = model_outputs["logits"][0]
if self.framework == "pt":
outputs = outputs.cpu().numpy()
else:
outputs = outputs.numpy()
if function_to_apply == ClassificationFunction.SIGMOID:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment