mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix xtransformers bug

Browse Source
This commit is contained in:
James Betker 2022-04-08 09:12:46 -06:00
parent 048f6f729a
commit 423293e518
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
codes/models/lucidrains/x_transformers.py
View File

@ -1180,7 +1180,9 @@ class TransformerWrapper(nn.Module):
if use_cache:
res.append(intermediates.past_key_values)
return res
if len(res) > 1:
return tuple(res)
return res[0]
class ContinuousTransformerWrapper(nn.Module):
@ -1241,7 +1243,9 @@ class ContinuousTransformerWrapper(nn.Module):
if use_cache:
res.append(intermediates.past_key_values)
return tuple(res)
if len(res) > 1:
return tuple(res)
return res[0]
class XTransformer(nn.Module):