abap-nous-hermes / README.md
smjain's picture
Update README.md
05dfef8
|
raw
history blame
688 Bytes
metadata
license: apache-2.0
datasets:
  - smjain/abap
language:
  - en

This model is fine tuned on a very small ABAP dataset . Have used NousResearch/Llama-2-7b-chat-hf as the base model.

Sample code

from transformers import pipeline from transformers import AutoModelForCausalLM, AutoTokenizer

model_path = "smjain/abap-nous-hermes"

model = AutoModelForCausalLM.from_pretrained(model_path)

tokenizer = AutoTokenizer.from_pretrained('NousResearch/llama-2-7b-chat-hf')

prompt = "Write a sample ABAP report" # change to your desired prompt

gen = pipeline('text-generation', model=model, tokenizer=tokenizer,max_new_tokens=256)

result = gen(prompt)

print(result[0]['generated_text'])