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

Fix I thought wasn't needed since it literally worked without it earlier

Browse Source
This commit is contained in:
mrq 2023-02-17 20:41:20 +00:00
parent 535549c3f3
commit e3e8801e5f
1 changed files with 10 additions and 0 deletions

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

@ -1,9 +1,19 @@
import re import re
import torch import torch
from torch import nn from torch import nn
# for god knows why it cannot "see" performer_pytorch
import os
import sys
prev_sys = copy(sys.path)
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_'