Spaces:

Tonic
/

Math

Running on Zero

Tonic commited on 7 days ago

Commit

b8790bd

unverified ·

1 Parent(s): 9dbcbb8

switch to smaller model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -100,7 +100,7 @@ examples = [
     [unimath4, additional_info_prompt, 2500]
 ]
-model_name = "AI-MO/Kimina-Prover-72B"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto", trust_remote_code=True)

     [unimath4, additional_info_prompt, 2500]
 ]
+model_name = "AI-MO/Kimina-Prover-Distill-8B"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto", trust_remote_code=True)