Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add instructions for playing with qlora model to colab example #1290

Merged
merged 2 commits into from
Feb 21, 2024

Conversation

jaredpalmer
Copy link
Contributor

@jaredpalmer jaredpalmer commented Feb 13, 2024

Description

Add instructions for launching a gradio app on google colab to play with finetuned model

Motivation and Context

How has this been tested?

Screenshots (if appropriate)

CleanShot 2024-02-13 at 14 43 57@2x

Types of changes

Social Handles (Optional)

@JohanWork
Copy link
Contributor

JohanWork commented Feb 16, 2024

For me this example dosen't work on the free tier. Getting 'RuntimeError: FlashAttention only supports Ampere GPUs or newer.' Would be nice to have a setup that work for everyone without having to have a premium account :) Changing to

!accelerate launch -m axolotl.cli.inference /content/test_axolotl.yaml \
    --qlora_model_dir="./qlora-out" --gradio

for example work for me.
@jaredpalmer
For reference I am using a T4

@NanoCode012
Copy link
Collaborator

I think @JohanWork 's suggestion is nice. It makes sense also to reuse the same config from training instead of another yaml.

Co-authored-by: JohanWork <39947546+JohanWork@users.noreply.github.com>
@NanoCode012 NanoCode012 merged commit 6ab69ec into axolotl-ai-cloud:main Feb 21, 2024
djsaunde pushed a commit that referenced this pull request Dec 17, 2024
* Add instructions for playing with qlora model to colab example

* Update examples/colab-notebooks/colab-axolotl-example.ipynb

Co-authored-by: JohanWork <39947546+JohanWork@users.noreply.github.com>

---------

Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
Co-authored-by: JohanWork <39947546+JohanWork@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants