Upload folder using huggingface_hub
Browse files- Test_RAG.py +3 -2
Test_RAG.py
CHANGED
@@ -181,12 +181,13 @@ def convert_to_int4():
|
|
181 |
int4_compression_args = " --group-size {} --ratio {}".format(model_compression_params["group_size"], model_compression_params["ratio"])
|
182 |
if model_compression_params["sym"]:
|
183 |
int4_compression_args += " --sym"
|
184 |
-
|
|
|
185 |
int4_compression_args += " --awq --dataset wikitext2 --num-samples 128"
|
186 |
export_command_base += int4_compression_args
|
187 |
if remote_code:
|
188 |
export_command_base += " --trust-remote-code"
|
189 |
-
export_command = export_command_base + " " + str(int4_model_dir)
|
190 |
|
191 |
|
192 |
|
|
|
181 |
int4_compression_args = " --group-size {} --ratio {}".format(model_compression_params["group_size"], model_compression_params["ratio"])
|
182 |
if model_compression_params["sym"]:
|
183 |
int4_compression_args += " --sym"
|
184 |
+
print("updated")
|
185 |
+
if enable_awq:
|
186 |
int4_compression_args += " --awq --dataset wikitext2 --num-samples 128"
|
187 |
export_command_base += int4_compression_args
|
188 |
if remote_code:
|
189 |
export_command_base += " --trust-remote-code"
|
190 |
+
# export_command = export_command_base + " " + str(int4_model_dir)
|
191 |
|
192 |
|
193 |
|