Fix later
This commit is contained in:
parent
e3e8801e5f
commit
5ecf7da881
|
@ -6,14 +6,11 @@ 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 autoregressive_wrapper import AutoregressiveWrapper
|
||||
|
||||
sys.path = prev_sys
|
||||
|
||||
ENC_PREFIX = 'enc_'
|
||||
DEC_PREFIX = 'dec_'
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user