smaller model
Browse files- 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-
|
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-70b-hf",
|
20 |
messages=[
|
21 |
{"role": "system", "content": system_message},
|
22 |
{"role": "user", "content": propmpt},
|