Update README.md

README.md

@@ -33,6 +33,22 @@ This design enables:
 - Fast adaptation to downstream tasks
 - Improved few-shot and full-data performance
 
+Usage: 
+
+Before running the following code, we need to import the modules from https://github.com/baoguangsheng/faast 
+
+```
+tokenizer = AutoTokenizer.from_pretrained(args.model_path, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(args.model_path, trust_remote_code=True)
+
+fewshot_samples = ['sample 1', 'sample 2', ...]
+inputs = tokenizer(fewshot_samples, return_tensors="pt", padding=True)
+
+model.reset_projection() # clear existing fast weights
+model.learn(**inputs)  # learn new fast weights
+model.generate(...)  # do the task using the learned fast weights
+```
+
 ## Training Details
 
 - **Base model:** Qwen2.5-3B-Instruct