amaye15 commited on
Commit
767a789
·
verified ·
1 Parent(s): 1afcb0c

Update modeling_autoencoder.py to support relative import for HF dynamic modules

Browse files
Files changed (1) hide show
  1. modeling_autoencoder.py +4 -1
modeling_autoencoder.py CHANGED
@@ -13,7 +13,10 @@ from transformers import PreTrainedModel
13
  from transformers.modeling_outputs import BaseModelOutput
14
  from transformers.utils import ModelOutput
15
 
16
- from configuration_autoencoder import AutoencoderConfig
 
 
 
17
 
18
 
19
  class NeuralScaler(nn.Module):
 
13
  from transformers.modeling_outputs import BaseModelOutput
14
  from transformers.utils import ModelOutput
15
 
16
+ try:
17
+ from .configuration_autoencoder import AutoencoderConfig # when loaded via HF dynamic module
18
+ except Exception:
19
+ from configuration_autoencoder import AutoencoderConfig # local usage
20
 
21
 
22
  class NeuralScaler(nn.Module):