AndrewMayesPrezzee commited on
Commit
5eda1f5
·
1 Parent(s): 5b68b61

Feat - Relative & Absolute Imports

Browse files
configuration_autoencoder.py CHANGED
@@ -5,6 +5,12 @@ Autoencoder configuration for Hugging Face Transformers.
5
  from transformers import PretrainedConfig
6
  from typing import List, Optional
7
 
 
 
 
 
 
 
8
 
9
  class AutoencoderConfig(PretrainedConfig):
10
  """
 
5
  from transformers import PretrainedConfig
6
  from typing import List, Optional
7
 
8
+ # Support both package-relative and flat import in HF remote code context
9
+ try:
10
+ from . import __version__ as _pkg_version # type: ignore
11
+ except Exception: # pragma: no cover
12
+ _pkg_version = None
13
+
14
 
15
  class AutoencoderConfig(PretrainedConfig):
16
  """
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):