echarlaix HF staff commited on
Commit
6c76373
·
1 Parent(s): d4c3e31

update description

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -107,7 +107,7 @@ def quantize_model(
107
  quant_method=quant_method,
108
  dataset=None if quant_method=="default" else calibration_dataset,
109
  ratio=1.0 if is_int8 else ratio,
110
- num_samples=50,
111
  )
112
 
113
  api = HfApi(token=oauth_token.token)
@@ -195,7 +195,7 @@ def quantize_model(
195
  path_in_repo="README.md",
196
  repo_id=new_repo_id,
197
  )
198
- return f"This model was successfully quantized, find it under your repo {new_repo_url}'"
199
  finally:
200
  shutil.rmtree(folder, ignore_errors=True)
201
  except Exception as e:
@@ -261,7 +261,7 @@ private_repo = gr.Checkbox(
261
  overwritte = gr.Checkbox(
262
  value=False,
263
  label="Overwrite repo content",
264
- info="Push files on existing repo potentially overwriting existing files",
265
  )
266
  interface = gr.Interface(
267
  fn=quantize_model,
 
107
  quant_method=quant_method,
108
  dataset=None if quant_method=="default" else calibration_dataset,
109
  ratio=1.0 if is_int8 else ratio,
110
+ num_samples=None if quant_method=="default" else 50,
111
  )
112
 
113
  api = HfApi(token=oauth_token.token)
 
195
  path_in_repo="README.md",
196
  repo_id=new_repo_id,
197
  )
198
+ return f"This model was successfully quantized, find it under your repo {new_repo_url}"
199
  finally:
200
  shutil.rmtree(folder, ignore_errors=True)
201
  except Exception as e:
 
261
  overwritte = gr.Checkbox(
262
  value=False,
263
  label="Overwrite repo content",
264
+ info="Enable pushing files on existing repo, potentially overwriting existing files",
265
  )
266
  interface = gr.Interface(
267
  fn=quantize_model,