riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 18:51:14 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fixed shape issue for T5 tracing (#11742)

Browse Source 
Co-authored-by: Michael Benayoun <michael@huggingface.co>
This commit is contained in:
Michael Benayoun 2021-05-17 12:17:31 +02:00 committed by GitHub
parent 0fc56df5fb
commit a0531c8a24
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_fx_utils.py
View File

@ -68,6 +68,8 @@ class HFProxy(Proxy):
if self.tracer.num_choices <= 0: if self.tracer.num_choices <= 0:
raise ValueError("num_choices must be given to the CustomTracer for MultipleChoice tasks.") raise ValueError("num_choices must be given to the CustomTracer for MultipleChoice tasks.")
shape = shape[:1] + [self.tracer.num_choices] + shape[1:] shape = shape[:1] + [self.tracer.num_choices] + shape[1:]
elif "hidden_states.s" in code_context:
shape = shape + [self.tracer.root.config.hidden_size]
else: else:
# Default case: # Default case:
# - If self.size is called for an unpacking, retrieves the corresponding unpacking # - If self.size is called for an unpacking, retrieves the corresponding unpacking