tomaarsen

# See the Documentation for more information: https://cubiedev.github.io/TheNounProjectAPI
# Install Module using "pip install TheNounProjectAPI".
# This sample works for version >= 1.0.5
from TheNounProjectAPI import API

if __name__ == "__main__":
    # API Key and Secret from https://api.thenounproject.com/getting_started.html#creating-an-api-key
    key = "<my api key>"
    secret = "<my api secret>"

tomaarsen

import functools
from dataclasses import dataclass
from typing import Callable

from datasets import load_dataset
from sentence_transformers.losses import CosineSimilarityLoss
from torch import nn

import wandb
from setfit import SetFitModel, SetFitTrainer, sample_dataset

tomaarsen

# Clone the full original repo into a local directory
git clone https://huggingface.co/datasets/ought/raft-submission local-raft-submission
cd local-raft-submission

# Fetch the PR and place it in a branch called pr/3
git fetch origin pr/3:pr/3

# Checkout the new branch
git checkout pr/3

tomaarsen

from datasets import load_dataset, concatenate_datasets
from transformers import TrainingArguments
from span_marker import SpanMarkerModel, Trainer


def main() -> None:
    # Load the dataset, ensure "tokens" and "ner_tags" columns, and get a list of labels
    dataset = load_dataset("midas/inspec", "extraction")
    dataset = dataset.rename_columns({"document": "tokens", "doc_bio_tags": "ner_tags"})
    # Map string labels to integer labels instead

tomaarsen

from typing import Any, Dict, List

from span_marker import SpanMarkerModel


class EndpointHandler:
    def __init__(self, model_id: str) -> None:
        self.model = SpanMarkerModel.from_pretrained(model_id)
        # Try to place it on CUDA, do nothing if it fails
        self.model.try_cuda()

tomaarsen

from transformers import AutoTokenizer, SinkCache, LlamaForCausalLM, TextStreamer
import torch

tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf")
model = LlamaForCausalLM.from_pretrained(
    "meta-llama/Llama-2-7b-hf", device_map="auto", torch_dtype=torch.float16
)

inputs = tokenizer(["Vaswani et al. (2017) introduced the Transformers"], return_tensors="pt").to(model.device)
cache = SinkCache(window_length=256, num_sink_tokens=4)

tomaarsen

from functools import partial
import datasets
from sentence_transformers import (
    SentenceTransformer,
    evaluation,
)
from torch.nn import functional as F

stsb = datasets.load_dataset("mteb/stsbenchmark-sts", split="test")

tomaarsen

tomaarsen

# requires sentence_transformers>=3.2.0
from sentence_transformers import SentenceTransformer, export_optimized_onnx_model, export_dynamic_quantized_onnx_model

# The model to export to ONNX (+ optimize, quantize), OpenVINO
model_id = "mixedbread-ai/mxbai-embed-large-v1"
# Where to save the exported models locally
output_dir = model_id.replace("/", "-")

onnx_model = SentenceTransformer(model_id, backend="onnx", model_kwargs={"export": True})
onnx_model.save_pretrained(output_dir)