deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix later

Browse Source
This commit is contained in:
mrq 2023-02-17 20:49:29 +00:00
parent e3e8801e5f
commit 5ecf7da881
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/models/lucidrains/performer/performer_enc_dec.py
View File

@ -6,14 +6,11 @@ from torch import nn
# for god knows why it cannot "see" performer_pytorch # for god knows why it cannot "see" performer_pytorch
import os import os
import sys import sys
prev_sys = copy(sys.path)
sys.path.insert(0, os.path.dirname(os.path.realpath(__file__))) sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)))
from performer_pytorch import PerformerLM from performer_pytorch import PerformerLM
from autoregressive_wrapper import AutoregressiveWrapper from autoregressive_wrapper import AutoregressiveWrapper
sys.path = prev_sys
ENC_PREFIX = 'enc_' ENC_PREFIX = 'enc_'
DEC_PREFIX = 'dec_' DEC_PREFIX = 'dec_'