Reduce memory requirement on export_llama tests with no params (#11339)

Summary:


Reviewed By: larryliu0820

Differential Revision: D75498713

Author: jackzhxng

examples/models/llama/model_args.py

@@ -5,7 +5,7 @@
 @dataclass
 class ModelArgs:
     dim: int = 4096
-    n_layers: int = 8
+    n_layers: int = 1
     n_heads: int = 32
     n_kv_heads: Optional[int] = None
     vocab_size: int = 512  # Arbitrary value, should be defined later by tokenizer.