Update step_ablation generate fallback

analysis/step_ablation.py

@@ -448,7 +448,11 @@ def evaluate_all_models(models: Dict[int, object],
                 src = src.unsqueeze(0)
 
             with torch.no_grad():
-                out = model.model.generate_cached(src.to(device))
+                if hasattr(model, "model") and hasattr(model.model, "generate_cached"):
+                    out = model.model.generate_cached(src.to(device))
+                else:
+                    # Fallback for wrappers that only expose top-level generate.
+                    out = model.generate(src.to(device))
 
             ids = [x for x in out[0].tolist() if x > 4]
             pred = tgt_tokenizer.decode(ids).strip()
@@ -579,4 +583,4 @@ def run_task4(models, src_list, ref_list, tgt_tokenizer):
     with open("analysis/outputs/task4_report.txt", "w") as f:
         f.write(f"Optimal diffusion steps = {knee_T}\n")
 
-    return knee_T
+    return knee_T