add cleanup

tests/peft_test.sh

@@ -2,6 +2,10 @@
 # set -x
 set -e
 
+cleanup() {
+    rm -rf ~/.cache/flexflow/debug
+}
+
 # Cd into directory holding this script
 cd "${BASH_SOURCE[0]%/*}/.."
 
@@ -11,6 +15,9 @@ if [[ "$HUGGINGFACE_TOKEN" != "none" ]]; then
     huggingface-cli login --token "$HUGGINGFACE_TOKEN"
 fi
 
+# Clean up before test (just in case)
+cleanup
+
 # Create test prompt file
 mkdir -p ./inference/prompt
 echo '["Two things are infinite: "]' > ./inference/prompt/peft.json
@@ -54,3 +61,6 @@ python ./tests/peft/peft_alignment_test.py -tp 2
 echo ""
 echo "PEFT tests passed!"
 echo ""
+
+# Cleanup after the test
+cleanup