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

This commit is contained in:
mrq 2023-02-17 20:41:20 +00:00
parent 535549c3f3
commit e3e8801e5f

View File

@ -1,9 +1,19 @@
import re
import torch
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_'