I don't know what I created it for, lol

7 Labels

label meaning
move_forward move straight when the path is clear
fly take off / fly
turn_left / turn_right turn left/right when encountering an obstacle on the right/left
spin_360 rotate and re-scan when disoriented
evade_threat evade threats (shadows, swatters, predators)
land_on_food land where a food scent is detected

Usage

1) GGUF F16 / Q4_K_M with llama.cpp / Ollama / LM Studio:

# Download
hf download mondk/fly-brain fly-behavior-classifier.F16.gguf
# or: huggingface-cli download mondk/fly-brain fly-behavior-classifier.F16.gguf

# Ollama
cat > Modelfile <<'EOF'
FROM ./fly-behavior-classifier.F16.gguf
EOF
ollama create fly-brain -f Modelfile
ollama run fly-brain

# llama.cpp (embedding/classifier โ€” this GGUF is a DistilBERT classifier, not a chat LM,
# so don't expect it to chat like an LLM; the file is primarily for loading via llama.cpp/LM Studio.
# The actual classification demo still runs via Transformers as shown above)
./llama.cpp/build/bin/llama-cli -m fly-behavior-classifier.F16.gguf -p "turn_left"

2) Space: https://huggingface.co/spaces/mondk/fly-brain

  • fly-behavior-classifier.F16.gguf (135 MB, F16) โ€” Full GGUF
  • fly-behavior-classifier.Q4_K_M.gguf (47.4 MB, 4-bit) โ€” Quantized GGUF, faster loading
  • fly-brain/ โ€” FlyWire LIF simulation code + 3D viewer (not directly related to the classifier; included for reference to the actual connectome)

ty

Downloads last month
69
GGUF
Model size
67M params
Architecture
bert
Hardware compatibility
Log In to add your hardware

4-bit

16-bit

Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Space using mondk/fly-brain 1