File size: 1,207 Bytes
eaeea8f
 
 
 
 
 
 
 
 
 
 
15a4381
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
"""
Hugging Face Transformers Integration for Q-TensorFormer.

Provides first-class Hugging Face ecosystem compatibility:
  - QTensorFormerConfig (inherits PretrainedConfig)
  - QTensorFormerForCausalLM (inherits PreTrainedModel, GenerationMixin)
  - AutoConfig & AutoModelForCausalLM registration
  - Generation support with past_key_values and Adaptive KV Cache
  - save_pretrained and from_pretrained serialization
"""

import sys
import os

# Re-export authoritative root implementations
from configuration_qtensorformer import QTensorFormerConfig, factorize_dim
from modeling_qtensorformer import (
    QTensorFormerForCausalLM,
    QTensorFormerModel,
    QTensorFormerPreTrainedModel,
    QTensorFormerBlock,
    QTensorFormerAttention,
    QTensorFormerMLP,
    TensorTrainLinear,
    AdaptiveKVCache,
    PIDResourceController,
    QTensorFormerRMSNorm,
)

__all__ = [
    "QTensorFormerConfig",
    "QTensorFormerForCausalLM",
    "QTensorFormerModel",
    "QTensorFormerPreTrainedModel",
    "QTensorFormerBlock",
    "QTensorFormerAttention",
    "QTensorFormerMLP",
    "TensorTrainLinear",
    "AdaptiveKVCache",
    "PIDResourceController",
    "QTensorFormerRMSNorm",
    "factorize_dim",
]