File size: 624 Bytes
d53adc9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# Generated by ml.integrations.export.runtime_packager.write_remote_code_bundle.
# Exported for HuggingFace trust_remote_code loading.
# This file is intentionally self-contained.

"""Remote-code naming shared by the HF adapter and export bundle."""

from __future__ import annotations

HF_MODELING_MODULE = "modeling_sophia"
HF_CONFIG_CLASS = "SophiaConfig"
HF_CAUSAL_LM_CLASS = "SophiaForCausalLM"
HF_SUPPORT_FILENAME = "hf_support.py"
HF_RUNTIME_FILENAME = "sophia_runtime.py"

__all__ = [
    "HF_CAUSAL_LM_CLASS",
    "HF_CONFIG_CLASS",
    "HF_MODELING_MODULE",
    "HF_RUNTIME_FILENAME",
    "HF_SUPPORT_FILENAME",
]