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 |
|