actor/memory_augment_config.json

{
  "format": "jamel_memory_augmented_lm",
  "format_version": 1,
  "base_model_name_or_path": ".",
  "self_contained": true,
  "memory_hidden_size": 2048,
  "class_name": "FSDPMemoryAugmentedCausalLM"
}