236 lines
6.5 KiB
Python
236 lines
6.5 KiB
Python
"""Public contracts for the framework-independent schedule core."""
|
|
|
|
from comfyui_sigmax.core.artifacts import (
|
|
NUMERICAL_SCHEDULE_SCHEMA,
|
|
SCHEDULE_ARTIFACT_ENVELOPE_SCHEMA,
|
|
SCHEDULE_ARTIFACT_SCHEMA,
|
|
ArtifactBuildMetadata,
|
|
ArtifactField,
|
|
ScheduleArtifact,
|
|
TypedArtifactValue,
|
|
build_schedule_artifact,
|
|
deserialize_schedule_artifact,
|
|
serialize_schedule_artifact,
|
|
)
|
|
from comfyui_sigmax.core.base_grids import (
|
|
flowmatch_reciprocal_step_grid,
|
|
krea_reciprocal_step_grid,
|
|
linear_endpoint_grid,
|
|
)
|
|
from comfyui_sigmax.core.capabilities import (
|
|
CapabilityCompatibilityError,
|
|
CapabilityDimension,
|
|
CompatibilityDecision,
|
|
CompatibilityLevel,
|
|
CompatibilityReason,
|
|
ExecutionBehavior,
|
|
ExecutionFeatureRequest,
|
|
ModelCapabilities,
|
|
NoiseOwnership,
|
|
PredictionType,
|
|
ProfileCapabilities,
|
|
SamplerCapabilities,
|
|
SamplerState,
|
|
TerminalRequirement,
|
|
TerminalSigma,
|
|
evaluate_compatibility,
|
|
require_compatible,
|
|
)
|
|
from comfyui_sigmax.core.execution_receipts import (
|
|
EXECUTION_RECEIPT_ENVELOPE_SCHEMA,
|
|
EXECUTION_RECEIPT_SCHEMA,
|
|
PORTABLE_EXECUTION_BUNDLE_SCHEMA,
|
|
ExecutionComponent,
|
|
ExecutionHost,
|
|
ExecutionReceipt,
|
|
ExecutionReceiptMetadata,
|
|
ExecutionRngOwnership,
|
|
ExecutionStatus,
|
|
PortableExecutionBundle,
|
|
build_execution_receipt,
|
|
deserialize_execution_receipt,
|
|
deserialize_portable_execution_bundle,
|
|
serialize_execution_receipt,
|
|
serialize_portable_execution_bundle,
|
|
)
|
|
from comfyui_sigmax.core.fingerprints import (
|
|
build_numerical_projection,
|
|
canonical_projection_bytes,
|
|
construction_fingerprint,
|
|
float_to_ieee_hex,
|
|
numerical_fingerprint,
|
|
)
|
|
from comfyui_sigmax.core.reports import (
|
|
SCHEDULE_COMPARISON_REPORT_ENVELOPE_SCHEMA,
|
|
SCHEDULE_COMPARISON_REPORT_SCHEMA,
|
|
SCHEDULE_REPORT_ENVELOPE_SCHEMA,
|
|
SCHEDULE_REPORT_SCHEMA,
|
|
ScheduleComparisonReport,
|
|
ScheduleReport,
|
|
build_schedule_comparison_report,
|
|
build_schedule_report,
|
|
build_schedule_report_from_bundle,
|
|
deserialize_schedule_comparison_report,
|
|
deserialize_schedule_report,
|
|
serialize_schedule_comparison_report,
|
|
serialize_schedule_report,
|
|
)
|
|
from comfyui_sigmax.core.request_result import (
|
|
BaseGridSpec,
|
|
EvidenceLevel,
|
|
OverrideRecord,
|
|
Provenance,
|
|
ScheduleInputs,
|
|
ScheduleRequest,
|
|
ScheduleResult,
|
|
SliceSpec,
|
|
TerminalPolicy,
|
|
)
|
|
from comfyui_sigmax.core.safetensors_header import (
|
|
SAFETENSORS_HEADER_MAX_BYTES,
|
|
SafetensorsHeader,
|
|
SafetensorsHeaderError,
|
|
SafetensorsHeaderReason,
|
|
SafetensorsTensorDescriptor,
|
|
read_safetensors_header,
|
|
)
|
|
from comfyui_sigmax.core.schedule_contracts import (
|
|
ScheduleContractError,
|
|
ScheduleOwnership,
|
|
SigmaDomain,
|
|
TransformContract,
|
|
TransformStage,
|
|
require_single_ownership,
|
|
validate_transform_chain,
|
|
)
|
|
from comfyui_sigmax.core.shifts import (
|
|
direct_ratio_shift,
|
|
exponential_mu_shift,
|
|
no_shift,
|
|
)
|
|
from comfyui_sigmax.core.terminal_slicing import (
|
|
apply_terminal_policy,
|
|
denoise_construction_steps,
|
|
slice_denoise_tail,
|
|
slice_step_range,
|
|
)
|
|
from comfyui_sigmax.core.validation import validate_sigma_schedule
|
|
from comfyui_sigmax.core.workflow_metadata import (
|
|
WorkflowArtifactReference,
|
|
WorkflowHostRequirement,
|
|
WorkflowMetadata,
|
|
WorkflowReceiptReference,
|
|
WorkflowRequirement,
|
|
attach_workflow_metadata,
|
|
deserialize_workflow_metadata,
|
|
detach_workflow_metadata,
|
|
extract_workflow_metadata,
|
|
serialize_workflow_metadata,
|
|
)
|
|
|
|
__all__ = [
|
|
"EXECUTION_RECEIPT_ENVELOPE_SCHEMA",
|
|
"EXECUTION_RECEIPT_SCHEMA",
|
|
"NUMERICAL_SCHEDULE_SCHEMA",
|
|
"PORTABLE_EXECUTION_BUNDLE_SCHEMA",
|
|
"SAFETENSORS_HEADER_MAX_BYTES",
|
|
"SCHEDULE_ARTIFACT_ENVELOPE_SCHEMA",
|
|
"SCHEDULE_ARTIFACT_SCHEMA",
|
|
"SCHEDULE_COMPARISON_REPORT_ENVELOPE_SCHEMA",
|
|
"SCHEDULE_COMPARISON_REPORT_SCHEMA",
|
|
"SCHEDULE_REPORT_ENVELOPE_SCHEMA",
|
|
"SCHEDULE_REPORT_SCHEMA",
|
|
"ArtifactBuildMetadata",
|
|
"ArtifactField",
|
|
"BaseGridSpec",
|
|
"CapabilityCompatibilityError",
|
|
"CapabilityDimension",
|
|
"CompatibilityDecision",
|
|
"CompatibilityLevel",
|
|
"CompatibilityReason",
|
|
"EvidenceLevel",
|
|
"ExecutionBehavior",
|
|
"ExecutionComponent",
|
|
"ExecutionFeatureRequest",
|
|
"ExecutionHost",
|
|
"ExecutionReceipt",
|
|
"ExecutionReceiptMetadata",
|
|
"ExecutionRngOwnership",
|
|
"ExecutionStatus",
|
|
"ModelCapabilities",
|
|
"NoiseOwnership",
|
|
"OverrideRecord",
|
|
"PortableExecutionBundle",
|
|
"PredictionType",
|
|
"ProfileCapabilities",
|
|
"Provenance",
|
|
"SafetensorsHeader",
|
|
"SafetensorsHeaderError",
|
|
"SafetensorsHeaderReason",
|
|
"SafetensorsTensorDescriptor",
|
|
"SamplerCapabilities",
|
|
"SamplerState",
|
|
"ScheduleArtifact",
|
|
"ScheduleComparisonReport",
|
|
"ScheduleContractError",
|
|
"ScheduleInputs",
|
|
"ScheduleOwnership",
|
|
"ScheduleReport",
|
|
"ScheduleRequest",
|
|
"ScheduleResult",
|
|
"SigmaDomain",
|
|
"SliceSpec",
|
|
"TerminalPolicy",
|
|
"TerminalRequirement",
|
|
"TerminalSigma",
|
|
"TransformContract",
|
|
"TransformStage",
|
|
"TypedArtifactValue",
|
|
"WorkflowArtifactReference",
|
|
"WorkflowHostRequirement",
|
|
"WorkflowMetadata",
|
|
"WorkflowReceiptReference",
|
|
"WorkflowRequirement",
|
|
"apply_terminal_policy",
|
|
"attach_workflow_metadata",
|
|
"build_execution_receipt",
|
|
"build_numerical_projection",
|
|
"build_schedule_artifact",
|
|
"build_schedule_comparison_report",
|
|
"build_schedule_report",
|
|
"build_schedule_report_from_bundle",
|
|
"canonical_projection_bytes",
|
|
"construction_fingerprint",
|
|
"denoise_construction_steps",
|
|
"deserialize_execution_receipt",
|
|
"deserialize_portable_execution_bundle",
|
|
"deserialize_schedule_artifact",
|
|
"deserialize_schedule_comparison_report",
|
|
"deserialize_schedule_report",
|
|
"deserialize_workflow_metadata",
|
|
"detach_workflow_metadata",
|
|
"direct_ratio_shift",
|
|
"evaluate_compatibility",
|
|
"exponential_mu_shift",
|
|
"extract_workflow_metadata",
|
|
"float_to_ieee_hex",
|
|
"flowmatch_reciprocal_step_grid",
|
|
"krea_reciprocal_step_grid",
|
|
"linear_endpoint_grid",
|
|
"no_shift",
|
|
"numerical_fingerprint",
|
|
"read_safetensors_header",
|
|
"require_compatible",
|
|
"require_single_ownership",
|
|
"serialize_execution_receipt",
|
|
"serialize_portable_execution_bundle",
|
|
"serialize_schedule_artifact",
|
|
"serialize_schedule_comparison_report",
|
|
"serialize_schedule_report",
|
|
"serialize_workflow_metadata",
|
|
"slice_denoise_tail",
|
|
"slice_step_range",
|
|
"validate_sigma_schedule",
|
|
"validate_transform_chain",
|
|
]
|