Skip to main content

An LLM framework

Project description

Grace Hopper

🧠 Geniusrise

GitHub Workflow Status Codecov Codecov Codecov


About

Geniusrise is a modular, loosely-coupled AgentOps / MLOps framework designed for the era of Large Language Models, offering flexibility, inclusivity, and standardization in designing networks of AI agents.

It seamlessly integrates tasks, state management, data handling, and model versioning, all while supporting diverse infrastructures and user expertise levels. With its plug-and-play architecture, Geniusrise empowers teams to build, share, and deploy AI agent workflows across various platforms efficiently.

TLDR 🙄

1. Install geniusrise

pip install geniusrise
pip install geniusrise-huggingface

2. Create genius.yaml

version: '1'
bolts:
  HuggingFaceInstructionTuningBolt:
    name: 'hf-fine-tune-my-shit'
    method: fine_tune
    args:
      model_name: bert-base-uncased
      tokenizer_name: bert-base-uncased
      batches: 2
      hf_repo_id: my/repo
      token: 'hf_woohoo'
      commit_message: say hello to genius!
    input:
      type: batch
      args:
        bucket: geniusrise-test
        folder: my-shit
    output:
      type: batch
      args:
        bucket: geniusrise-test
        folder: my-model
    deploy:
      type: 'k8s'
      args:
        cluster_name: my-cluster
        namespace: geniusrise-huggingface
        labels: { 'needs': 'gpu' }
        cpu: 16
        memory: 50G
        storage: 250Gb
        gpu: 1

3. Copy data to s3

cat > data.jsonl <<- EOM
{"instruction": "instruction1", "output":"output1"}
{"instruction": "instruction2", "output":"output2"}
EOM

aws s3 cp data.jsonl s3://geniusrise-test/my-shit/

4. Fine tune

genius --yaml genius.yaml deploy

🙄 This was not even crux of the iceberg. Please see docs.

Links

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

geniusrise-0.0.12-py3-none-any.whl (100.9 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page