ONNX Model Zoo

mgoin/deepseek-coder-1.3b-instruct-pruned50-quant-ds

By mgoin · ↓ 2 · ♥ 0 · Updated 2023-11-07

What's in this repo

This page lists the files in mgoin/deepseek-coder-1.3b-instruct-pruned50-quant-ds, including model.onnx.

  • 1 ONNX models (.onnx files)
  • 7 files, including 4 .json, 1 .md, 1 .onnx
  • Last updated 2023-11-07

ONNX files

File Size Download
model.onnx — Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

DeepSparse One-Shot and Export of https://huggingface.co/deepseek-ai/deepseek-coder-1.3b-instruct

Prompt Template:

prompt = f"### Instruction:{input}### Response:"

Usage

from deepsparse import TextGeneration
model = TextGeneration(model="hf:mgoin/deepseek-coder-1.3b-instruct-pruned50-quant-ds")
print(model("#write a quick sort algorithm in python", max_new_tokens=200).generations[0].text)

"""
def quick_sort(arr):
    if len(arr) == 0:
        return arr
    else:
        pivot = arr[0]
        left = []
        right = []
        for i in range(1, len(arr)):
            if arr[i] < pivot:
                left.append(arr[i])
            else:
                right.append(arr[i])
        return quick_sort(left) + [pivot] + quick_sort(right)

print(quick_sort([3, 9, 2, 5, 4, 7, 1]))
print(quick_sort([1, 2, 3, 4, 5, 6]))
print(quick_sort([6, 5, 4, 3, 2, 1]))
"""

Model info

text-generation transformers
transformers onnx llama text-generation deepsparse conversational endpoints_compatible region:us
↓ 2 ♥ 0

Created: 2023-11-06

Updated: 2023-11-07

View on Hugging Face

Files (7)

.gitattributes
README.md
config.json
model.onnx ONNX
special_tokens_map.json
tokenizer.json
tokenizer_config.json

Related models