neuralmagic
/

MiniChat-3B-pruned50-quant-ds

Text Generation

Model card Files Files and versions Community

mwitiderrick commited on Dec 1, 2023

Commit

a608ae8

·

1 Parent(s): 507bf37

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -51,7 +51,7 @@ print(model(formatted_prompt, max_new_tokens=500).generations[0].text)
 ```bash
 git clone https://github.com/neuralmagic/sparseml
 pip install -e "sparseml[transformers]"
-python sparseml/src/sparseml/transformers/sparsification/obcq/obcq.py TinyLlama/TinyLlama-1.1B-Chat-v0.4 open_platypus --recipe recipe.yaml --save True
 python sparseml/src/sparseml/transformers/sparsification/obcq/export.py --task text-generation --model_path obcq_deployment
 cp deployment/model.onnx deployment/model-orig.onnx
 python onnx_kv_inject.py --input-file deployment/model-orig.onnx --output-file deployment/model.onnx

 ```bash
 git clone https://github.com/neuralmagic/sparseml
 pip install -e "sparseml[transformers]"
+python sparseml/src/sparseml/transformers/sparsification/obcq/obcq.py GeneZC/MiniChat-3B open_platypus --recipe recipe.yaml --save True
 python sparseml/src/sparseml/transformers/sparsification/obcq/export.py --task text-generation --model_path obcq_deployment
 cp deployment/model.onnx deployment/model-orig.onnx
 python onnx_kv_inject.py --input-file deployment/model-orig.onnx --output-file deployment/model.onnx