RicardoDominguez

"""Boundary value problem solver."""

import jax
import jax.numpy as jnp


# ------------------------------------------------------------------------------------------
# Linear solver for bordered almost block diagonal (BABD) systems
# ------------------------------------------------------------------------------------------
# Implementation as described in [1] Section 2.1 (structural orthogonal factorization).

RicardoDominguez

base_model: mistralai/Mistral-7B-v0.1
model_type: MistralForCausalLM
tokenizer_type: LlamaTokenizer
is_mistral_derived_model: true

load_in_8bit: false
load_in_4bit: false
strict: false

datasets:

RicardoDominguez

import datasets
import transformers
import vllm
from tqdm import tqdm

model_dir = 'meta-llama/Llama-2-7b-chat-hf'

# prompt from R1
system_prompt = "The user will ask you a question, and you should solve it. " \
        "You should first think about the reasoning process in the mind and then provide the user with the answer. " \