LlemmaFT / mse_ollama_timer.py
Zenos5's picture
Upload 24 files
766ea9e verified
raw
history blame contribute delete
575 Bytes
import jsonlines
import json
# import docx
# from deepeval.models import OllamaModel
import ollama
import time
#ollama run Hudson/llemma:7b
#deepeval set-ollama Hudson/llemma:7b
if __name__=="__main__":
# Initialize parser
tic = time.perf_counter()
ollama.generate(model="Hudson/llemma:7b", prompt="I need to determine the radius of convergence of the series $\\sum_{n=1}^\\infty a_nx^n$, where $a_n=a^n+b^n$ and $a,b$ are real numbers.")
# print(response)
toc = time.perf_counter()
print(f"Ran Llemma in {toc - tic:0.4f} seconds")