ramyaa1113 commited on
Commit
d01bb30
·
1 Parent(s): 492e40f

model inference changed

Browse files
Files changed (1) hide show
  1. translator/synth.py +2 -2
translator/synth.py CHANGED
@@ -24,7 +24,7 @@ def synthesize_cpp(parsed: dict, mapping: dict, prompt_context: str = "", model_
24
  return generate_with_openai(prompt, model="gpt-4o-mini")
25
  else:
26
  from .llm_client import generate_with_hf
27
- return generate_with_hf(prompt, model="mistralai/Mixtral-8x7B-Instruct")
28
 
29
 
30
  def synthesize_blueprint_python(parsed: dict, mapping: dict, model_backend="hf") -> str:
@@ -46,4 +46,4 @@ def synthesize_blueprint_python(parsed: dict, mapping: dict, model_backend="hf")
46
  return generate_with_openai(prompt, model="gpt-4o-mini")
47
  else:
48
  from .llm_client import generate_with_hf
49
- return generate_with_hf(prompt, model="mistralai/Mixtral-8x7B-Instruct")
 
24
  return generate_with_openai(prompt, model="gpt-4o-mini")
25
  else:
26
  from .llm_client import generate_with_hf
27
+ return generate_with_hf(prompt, model="mistralai/Mistral-7B-Instruct-v0.3")
28
 
29
 
30
  def synthesize_blueprint_python(parsed: dict, mapping: dict, model_backend="hf") -> str:
 
46
  return generate_with_openai(prompt, model="gpt-4o-mini")
47
  else:
48
  from .llm_client import generate_with_hf
49
+ return generate_with_hf(prompt, model="mistralai/Mistral-7B-Instruct-v0.3")