Anthonyg5005 commited on
Commit
52eb129
·
1 Parent(s): c7342b4

set correct repo yaml metadata

Browse files

instead of showing up as finetune, show up as quantized

exllamav2 scripts/auto-exl2-upload/exl2-quant.py CHANGED
@@ -177,6 +177,12 @@ if repo_exists(f"{whoami().get('name', None)}/{modelname}-exl2") == False:
177
  #create the markdown file
178
  print("Writing model card...")
179
  with open('./README.md', 'w') as file:
 
 
 
 
 
 
180
  file.write(f"# Exl2 quants for [{modelname}](https://huggingface.co/{repo_url})\n\n")
181
  file.write("## Automatically quantized using the auto quant script from [hf-scripts](https://huggingface.co/anthonyg5005/hf-scripts)\n\n")
182
  file.write("### BPW:\n\n")
 
177
  #create the markdown file
178
  print("Writing model card...")
179
  with open('./README.md', 'w') as file:
180
+ file.write('---\n')
181
+ file.write('library_name: ExLlama2\n')
182
+ file.write('pipeline_tag: text-generation\n')
183
+ file.write(f'base_model: {repo_url}\n')
184
+ file.write('base_model_relation: quantized\n')
185
+ file.write('---\n\n')
186
  file.write(f"# Exl2 quants for [{modelname}](https://huggingface.co/{repo_url})\n\n")
187
  file.write("## Automatically quantized using the auto quant script from [hf-scripts](https://huggingface.co/anthonyg5005/hf-scripts)\n\n")
188
  file.write("### BPW:\n\n")