smaller model

metadata_transformer.py

@@ -16,7 +16,7 @@ huggingface.prompt_builder = build_llama2_prompt
 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="mistralai/Mistral-7B-v0.1",
+        model="princeton-nlp/Sheared-LLaMA-1.3B",
         prompt=prompt,
         temperature=0.9,
         top_p=0.6,