Fix continue.

crates/ai00-core/src/run.rs

@@ -656,18 +656,13 @@ impl Runtime {
  slots[batch] = state;
  Ok(SlotResult::Fault(batch))
  }
- // continue from an existing slot; no need backing as well
- Some(SlotChoice::Continue(batch, len)) => {
+ // continue from an existing slot
+ Some(SlotChoice::Continue(batch, ..)) => {
  log::info!("continue at slot {}", batch);
  let checkout = self.checkout(context.request.state, &tokens, batch).await;
+ self.state.load(checkout.state, batch)?;
 
- // retrieve the last output from the cache
- assert!(checkout.prefix.len() <= len);
- if checkout.prefix.len() < len {
- self.state.load(checkout.state, batch)?;
- }
  let len = checkout.prefix.len();
-
  let state = SlotState::Wait(
  GenerateContext {
  prefix: Tokens(tokens[..len].to_vec()),