We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fc871a5 commit 7fcac78Copy full SHA for 7fcac78
tests/basic_correctness/test_cpu_offload.py
@@ -4,3 +4,5 @@
4
def test_cpu_offload():
5
compare_two_settings("meta-llama/Llama-2-7b-hf", [],
6
["--cpu-offload-gb", "4"])
7
+ compare_two_settings("nm-testing/llama7b-one-shot-2_4-w4a16-marlin24-t",
8
+ [], ["--cpu-offload-gb", "1"])
0 commit comments