1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4# Load model and tokenizer
5model_name = "StrangeSX/Saraa-8B-ORPO-AUNQA"
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForCausalLM.from_pretrained(
8 model_name,
9 torch_dtype=torch.float16,
10 device_map="auto"
11)
12
13# Example usage
14prompt = "Analyze this self-assessment report section for AUN-QA compliance:"
15inputs = tokenizer(prompt, return_tensors="pt")
16outputs = model.generate(**inputs, max_length=512, temperature=0.7)
17response = tokenizer.decode(outputs[0], skip_special_tokens=True)
18print(response)
1# Pull the model
2ollama pull strangex/saraa-8b-orpo-aunqa
3
4# Run inference
5ollama run strangex/saraa-8b-orpo-aunqa "What are the key criteria for AUN-QA standard 1?"
1@misc{saraa-8b-orpo-aunqa,
2 title={SARAA-8B-ORPO-AUNQA: Self-Assessment Report Analysis Assistant},
3 author={StrangeSX},
4 year={2024},
5 publisher={Hugging Face},
6 url={https://huggingface.co/StrangeSX/Saraa-8B-ORPO-AUNQA}
7}