Downloads · 30 days
0
zail-ai/Auramind
Auramind is a text generation model from zail-ai. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.
AuraMind is a collection of smartphone-optimized conversational AI models designed for dual-mode operation: Therapist Mode for emotional support and Assistant Mode for productivity coaching. Built on Gemma 2 270M arch…
Downloads · 30 days
0
Access
Public
Updated Aug 28, 2025
Repo size
—
Likes
0
Public
Click a slice to open those files.
.py17 KB · 54%
From the Hugging Face model README
AuraMind is a collection of smartphone-optimized conversational AI models designed for dual-mode operation: Therapist Mode for emotional support and Assistant Mode for productivity coaching. Built on Gemma 2 270M architecture and optimized for mobile deployment.
| Variant | Parameters | Memory | Speed | Repository |
|---|---|---|---|---|
| auramind_270 | 270M | ~680MB | 100-300ms | zail-ai/auramind-270m |
| auramind_180 | 180M | ~450MB | 80-200ms | zail-ai/auramind-180m |
| auramind_90 | 90M | ~225MB | 50-150ms | zail-ai/auramind-90m |
All variants run efficiently on modern smartphones with Android 8+ or iOS 12+.
Provides evidence-based emotional support and mental wellness guidance:
Therapeutic Approaches Integrated:
Delivers productivity coaching and task management support:
Productivity Frameworks Included:
from transformers import AutoTokenizer, AutoModelForCausalLM
import torch
# Load model (choose variant based on device)
model_name = "zail-ai/Auramind" # or specify variant: /auramind-270m
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
model_name,
torch_dtype=torch.float16, # Essential for mobile
device_map="auto",
low_cpu_mem_usage=True
)
def chat_with_auramind(message, mode="Assistant"):
"""Generate response in specified mode"""
prompt = f"<|start_of_turn|>user\n[{mode} Mode] {message}<|end_of_turn|>\n<|start_of_turn|>model\n"
inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
with torch.no_grad():
outputs = model.generate(
**inputs,
max_new_tokens=200,
temperature=0.7,
do_sample=True,
pad_token_id=tokenizer.eos_token_id,
repetition_penalty=1.1
)
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
return response.split("<|start_of_turn|>model\n")[-1].strip()
# Example usage
therapist_response = chat_with_auramind(
"I'm feeling anxious about my job interview tomorrow",
"Therapist"
)
assistant_response = chat_with_auramind(
"Help me organize my daily schedule more effectively",
"Assistant"
)
// Using PyTorch Mobile
Module module = LiteModuleLoader.load(assetFilePath(this, "auramind_mobile.ptl"));
// Inference
Tensor inputTensor = TensorImageUtils.bitmapToFloat32Tensor(bitmap);
Tensor outputTensor = module.forward(IValue.from(inputTensor)).toTensor();
// Using PyTorch Mobile
guard let module = TorchModule(fileAtPath: torchModelPath) else { return }
// Inference
let output = module.predict(input: inputTensor)
| Device | Variant | Inference Time | Memory Usage |
|---|---|---|---|
| iPhone 14 Pro | 270M | ~120ms | 680MB |
| Samsung Galaxy S23 | 270M | ~140ms | 680MB |
| Google Pixel 7 | 180M | ~90ms | 450MB |
| iPhone 12 | 180M | ~110ms | 450MB |
| Samsung Galaxy A54 | 90M | ~70ms | 225MB |
| OnePlus Nord | 90M | ~80ms | 225MB |
@model{auramind2025,
title={AuraMind: Smartphone Dual-Mode AI Companion for Mental Health and Productivity},
author={Zail AI},
year={2025},
url={https://huggingface.co/zail-ai/Auramind},
license={MIT}
}
This project is licensed under the MIT License - see the LICENSE file for details.
AuraMind - Empowering mental wellness and productivity through accessible AI technology.