rtferraz commited on
Commit
b86b1ee
·
verified ·
1 Parent(s): e881ea3

Update package init to v0.2.0 with model exports

Browse files
Files changed (1) hide show
  1. src/domain_tokenizer/__init__.py +6 -14
src/domain_tokenizer/__init__.py CHANGED
@@ -5,6 +5,7 @@ Core components:
5
  - schema: DomainSchema, FieldSpec, FieldType
6
  - tokenizers: DomainTokenizerBuilder, per-field tokenizers
7
  - schemas: Predefined schemas (FINANCE, ECOMMERCE, HEALTHCARE)
 
8
  """
9
 
10
  from .schema import DomainSchema, FieldSpec, FieldType
@@ -17,18 +18,9 @@ from .tokenizers.field_tokenizers import (
17
  MagnitudeBucketTokenizer,
18
  SignTokenizer,
19
  )
 
 
 
 
20
 
21
- __version__ = "0.1.0"
22
-
23
- __all__ = [
24
- "DomainSchema",
25
- "FieldSpec",
26
- "FieldType",
27
- "DomainTokenizerBuilder",
28
- "BaseFieldTokenizer",
29
- "SignTokenizer",
30
- "MagnitudeBucketTokenizer",
31
- "DiscreteNumericalTokenizer",
32
- "CalendarTokenizer",
33
- "CategoricalTokenizer",
34
- ]
 
5
  - schema: DomainSchema, FieldSpec, FieldType
6
  - tokenizers: DomainTokenizerBuilder, per-field tokenizers
7
  - schemas: Predefined schemas (FINANCE, ECOMMERCE, HEALTHCARE)
8
+ - models: DomainTransformerForCausalLM, PLR, JointFusion
9
  """
10
 
11
  from .schema import DomainSchema, FieldSpec, FieldType
 
18
  MagnitudeBucketTokenizer,
19
  SignTokenizer,
20
  )
21
+ from .models.configuration import DomainTransformerConfig
22
+ from .models.modeling import DomainTransformerForCausalLM, DomainTransformerModel
23
+ from .models.plr_embeddings import PeriodicLinearReLU
24
+ from .models.joint_fusion import JointFusionModel, DCNv2
25
 
26
+ __version__ = "0.2.0"