Views
No views yet

<think> and </think> traces, this model represents the core generalist version of the Athenea family, intended as a foundation for specialized reasoning variants.⚠️ Important Note: This model uses an abliterated (uncensored) base version, providing full expressive freedom and unrestricted output generation. Users are fully responsible for any use or content produced by the model. It is intended exclusively for research and experimentation purposes.
<think> blocksNote: Fine-tuning was performed using Kronos, Aquiles-ai’s proprietary enterprise fine-tuning system.
uv pip install transformers torch accelerate1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model = AutoModelForCausalLM.from_pretrained("Aquiles-ai/Athenea-4B-Thinking",
5 dtype=torch.bfloat16,
6 trust_remote_code=True,
7 device_map="auto",
8 attn_implementation="flash_attention_2") # Requires flash-attn
9
10# Without flash-attn:
11# model = AutoModelForCausalLM.from_pretrained("Aquiles-ai/Athenea-4B-Thinking",
12# dtype="auto",
13# device_map="auto"
14# )
15
16tokenizer = AutoTokenizer.from_pretrained("Aquiles-ai/Athenea-4B-Thinking", trust_remote_code=True)
17messages = [
18 {"role": "user", "content": "Hey, explain to me in simple terms how reinforcement learning works."}
19]
20
21inputs = tokenizer.apply_chat_template(
22 messages,
23 add_generation_prompt=True,
24 tokenize=True,
25 return_dict=True,
26 return_tensors="pt",
27).to('cuda')
28
29with torch.no_grad():
30 output = model.generate(
31 **inputs,
32 max_new_tokens=8092,
33 pad_token_id=tokenizer.eos_token_id,
34 eos_token_id=tokenizer.eos_token_id,
35 )
36
37# Decode and print the output
38print(tokenizer.decode(output[0], skip_special_tokens=True))1from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
2import torch
3from threading import Thread
4
5model = AutoModelForCausalLM.from_pretrained("Aquiles-ai/Athenea-4B-Thinking",
6 dtype=torch.bfloat16,
7 trust_remote_code=True,
8 device_map="auto",
9 attn_implementation="flash_attention_2")
10
11tokenizer = AutoTokenizer.from_pretrained("Aquiles-ai/Athenea-4B-Thinking", trust_remote_code=True)
12
13messages = [
14 {"role": "user", "content": "Hey, explain the difference between artificial intelligence, machine learning, and deep learning."}
15]
16
17inputs = tokenizer.apply_chat_template(
18 messages,
19 add_generation_prompt=True,
20 tokenize=True,
21 return_dict=True,
22 return_tensors="pt",
23).to('cuda')
24
25# Create the streamer
26streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
27
28# Build kwargs for generate
29generate_kwargs = dict(
30 **inputs,
31 max_new_tokens=8092,
32 pad_token_id=tokenizer.eos_token_id,
33 eos_token_id=tokenizer.eos_token_id,
34 streamer=streamer,
35)
36
37def _generate_thread(model, kwargs):
38 with torch.no_grad():
39 model.generate(**kwargs)
40
41thread = Thread(target=_generate_thread, args=(model, generate_kwargs))
42
43thread.start()
44
45for chunk in streamer:
46 print(chunk, end="", flush=True)1vllm serve Aquiles-ai/Athenea-4B-Thinking \
2 --host 0.0.0.0 \
3 --port 8000 \
4 --api-key dummyapikey \
5 --max-model-len=16384 \
6 --async-scheduling \
7 --gpu-memory-utilization=0.901from openai import OpenAI
2
3client = OpenAI(api_key="dummyapikey", base_url="http://127.0.0.1:8000/v1")
4
5stream = client.chat.completions.create(
6 model="Aquiles-ai/Athenea-4B-Thinking",
7 messages=[{
8 "role": "user",
9 "content": "Hey, tell me how a large language model like Llama or GPT is trained."
10 }],
11 max_tokens=8092,
12 stream=True
13)
14
15for chunk in stream:
16 if chunk.choices[0].delta.content:
17 print(chunk.choices[0].delta.content, end="", flush=True)