Fix I thought wasn't needed since it literally worked without it earlier
This commit is contained in:
parent
535549c3f3
commit
e3e8801e5f
|
@ -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_'
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user