Fast-dDrive / __init__.py
xiwenyoumu's picture
Initial Fast-dDrive 3B release
5e9a603 verified
"""Fast-dDrive HF release package.
Mirrors the layout of ``Efficient-Large-Model/Fast_dVLM_3B`` on the Hugging Face
Hub: a single :mod:`modeling` module that holds the model definition plus
inference-time decoding paths (Section Diffusion, Scaffold Spec, and
Scaffold Spec with multi-trajectory rollouts), and a :mod:`configuration`
module with the config classes.
Users normally load the model via::
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained(
"Efficient-Large-Model/Fast_dDrive", # or local path
trust_remote_code=True,
)
with the ``auto_map`` entry in :file:`config.json` pointing back to the classes
defined here.
"""
from .configuration import (
Fast_dDriveConfig,
Fast_dDriveTextConfig,
Fast_dDriveVisionConfig,
)
from .modeling import Fast_dDriveForConditionalGeneration
__all__ = [
"Fast_dDriveConfig",
"Fast_dDriveTextConfig",
"Fast_dDriveVisionConfig",
"Fast_dDriveForConditionalGeneration",
]