Fix later

pull/2/head
mrq 2023-02-17 20:49:29 +07:00
parent e3e8801e5f
commit 5ecf7da881
1 changed files with 0 additions and 3 deletions

@ -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_'