aquin.compute.capture Namespace Reference

Functions
CaptureBundle	build_minimal_capture_bundle (str prompt, Any model, str session_id="", int|None sae_layer=None)
str	upload_capture (CaptureBundle bundle, str session_id, str api_key, str base_url="https://api.aquin.app")

Function Documentation

build_minimal_capture_bundle()

CaptureBundle build_minimal_capture_bundle	(	str	prompt,
		Any	model,
		str	session_id = "",
		int | None	sae_layer = None )

Run a forward pass through model and collect top SAE features + logit lens.
Falls back to empty tensors if SAE weights are not available.

Definition at line 22 of file capture.py.

upload_capture()

str upload_capture	(	CaptureBundle	bundle,
		str	session_id,
		str	api_key,
		str	base_url = "https://api.aquin.app" )

Upload a CaptureBundle to Aquin Cloud. Returns capture_id.

Definition at line 109 of file capture.py.