Kjgarza commited on
Commit
ef6f242
·
1 Parent(s): cd2ab3e

smaller model

Browse files
Files changed (1) hide show
  1. metadata_transformer.py +1 -1
metadata_transformer.py CHANGED
@@ -16,7 +16,7 @@ You are a metadata schema translator. You translate metadata from one schema to
16
  def translate(schema_input, schema_target):
17
  propmpt = '"""{} \n Translate the schema metadata file above to the schema: {}"""'.format(schema_input, schema_target)
18
  response = huggingface.ChatCompletion.create(
19
- model="meta-llama/Llama-2-70b-hf",
20
  messages=[
21
  {"role": "system", "content": system_message},
22
  {"role": "user", "content": propmpt},
 
16
  def translate(schema_input, schema_target):
17
  propmpt = '"""{} \n Translate the schema metadata file above to the schema: {}"""'.format(schema_input, schema_target)
18
  response = huggingface.ChatCompletion.create(
19
+ model="meta-llama/Llama-2-7b-hf",
20
  messages=[
21
  {"role": "system", "content": system_message},
22
  {"role": "user", "content": propmpt},