Add test script and README for goldenbrick

ReaLLMASIC · Aug 23, 2024 · c4be2b1 · c4be2b1
1 parent f4d3b29
commit c4be2b1
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/tf_np_golden_gen/README.md b/tf_np_golden_gen/README.md
@@ -0,0 +1,5 @@
+# WIP Section
+
+This is for porting weights directly from the model into numpy for granular and
+actual integer inference tests and quantization verification.
+
diff --git a/tf_np_golden_gen/test_script.sh b/tf_np_golden_gen/test_script.sh
@@ -0,0 +1,4 @@
+#!/bin/bash
+
+python3 test.py --weights_path test.pkl --n_head 3 --n_layer 6 --n_embd 48
+
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,4 @@
		#!/bin/bash

		python3 test.py --weights_path test.pkl --n_head 3 --n_layer 6 --n_embd 48