JiakaiDu commited on
Commit
a4f90dd
1 Parent(s): 77ed57e

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
- if enable_awq.value:
 
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