|
AQIT 0.1.0
|
Functions | |
| Generator[dict, None, None] | run_steer_stream (str prompt, int feature_idx, float strength=20.0, str model_id="llama-3.2-1b", int max_new_tokens=200, float temperature=0.7) |
Variables | |
| DEVICE = resolve_compute_device() | |
| Generator[dict, None, None] run_steer_stream | ( | str | prompt, |
| int | feature_idx, | ||
| float | strength = 20.0, | ||
| str | model_id = "llama-3.2-1b", | ||
| int | max_new_tokens = 200, | ||
| float | temperature = 0.7 ) |