Update README.md
Browse files
README.md
CHANGED
@@ -105,7 +105,7 @@ quant_stage:
|
|
105 |
targets: ["Linear"]
|
106 |
"""
|
107 |
|
108 |
-
model_stub = "
|
109 |
model_name = model_stub.split("/")[-1]
|
110 |
|
111 |
device_map = calculate_offload_device_map(
|
|
|
105 |
targets: ["Linear"]
|
106 |
"""
|
107 |
|
108 |
+
model_stub = "meta-llama/Meta-Llama-3.1-8B-Instruct"
|
109 |
model_name = model_stub.split("/")[-1]
|
110 |
|
111 |
device_map = calculate_offload_device_map(
|