aquin.engine.tools.stubs Namespace Reference

Functions
None	_wire (str name, dict[str, Any] schema)

Variables
dict	_LORA_PARAMS

Function Documentation

_wire()

None _wire ( str name, dict[str, Any] schema )

protected

Definition at line 24 of file stubs.py.

References _wire().

Referenced by _wire().

Variable Documentation

_LORA_PARAMS

dict aquin.engine.tools.stubs._LORA_PARAMS

protected

Initial value:

=  {
    "rank": {"type": "number"},
    "alpha": {"type": "number"},
    "lr": {"type": "number"},
    "epochs": {"type": "number"},
    "dropout": {"type": "number"},
    "targetModules": {"type": "array", "items": {"type": "string"}},
    "warmupSteps": {"type": "number"},
    "gradClip": {"type": "number"},
    "weightDecay": {"type": "number"},
    "gradAccumSteps": {"type": "number"},
    "optimizer": {"type": "string", "enum": ["adamw", "sgd", "lion"]},
    "scheduler": {"type": "string", "enum": ["cosine", "linear", "constant"]},
    "maxSeqLen": {"type": "number"},
    "useQlora": {"type": "boolean"},
    "rows": {"type": "array", "items": {"type": "object"}},
}

Definition at line 461 of file stubs.py.