#pragma once

#include <string>
#include <vector>
#include <array>
#include <set>

using namespace std;

void sample_from_model(string &ckpt_path, vector<int> &prompt, vector<tuple<vector<int>,vector<int>>> control, int batch_size, float temperature);