File size: 786 Bytes
d81ed4d feafd16 0c9de8c d81ed4d 0c9de8c d81ed4d 0c9de8c feafd16 ce364f3 feafd16 3ffbd50 3e3f9c8 ce364f3 1069df6 ce364f3 3e3f9c8 358a050 9455d77 2597dbe a0d1546 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import os
from huggingface_hub import login
from easyllm.clients import huggingface
from easyllm.prompt_utils import build_llama2_prompt
TOKEN = os.environ.get("TOKEN")
login(token=TOKEN)
huggingface.prompt_builder = build_llama2_prompt
# system_message = """
# You are a metadata schema translator. You translate metadata from one schema to another.
# """
def translate(schema_input, schema_target):
prompt = '"""{} \n Translate the schema metadata file above to the schema: {}"""'.format(schema_input, schema_target)
response = huggingface.Completion.create(
model="princeton-nlp/Sheared-LLaMA-1.3B",
prompt=prompt,
temperature=0.9,
top_p=0.6,
max_tokens=250,
)
print(response)
return response['choices'][0]['text'] |