# ==========================================
# CELL 9: SAVE LORA ADAPTER
# ==========================================
# Run this AFTER cell_08

os.makedirs(OUTPUT_DIR, exist_ok=True)

adapter_path = os.path.join(OUTPUT_DIR, "adapter")
model.save_pretrained(adapter_path)
tokenizer.save_pretrained(adapter_path)

print(f"Adapter saved to: {adapter_path}")

# Calculate size
adapter_size_mb = sum(
    os.path.getsize(os.path.join(adapter_path, f))
    for f in os.listdir(adapter_path)
) / (1024 * 1024)

print(f"Adapter size: {adapter_size_mb:.1f} MB")
print(f"\nTo load this adapter later, use:")
print(f"  model.load_adapter('{adapter_path}')")