Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/cbh123/youtune

Fine tune SDXL on YouTube videos
https://github.com/cbh123/youtune

Last synced: 3 months ago
JSON representation

Fine tune SDXL on YouTube videos

Awesome Lists containing this project

README

        

# YouTune

![DALL·E 2023-10-26 19 39 10 - Photo of a sophisticated logo for 'YouTune' The design emphasizes the theme of tuning videos It incorporates a stylized film reel that intertwines w](https://github.com/cbh123/youtune/assets/14149230/808411aa-cecc-4735-a2dd-18344317601a)

[YouTune Video Walkthrough](https://www.loom.com/share/193fa040b8074f44bb5ddabd4dd42b01?sid=4b09aa1b-5cd6-4e4f-a538-d3d62cb1bdc0)

YouTune makes it really easy to:

- fine-tune SDXL on images from YouTube videos
- fine-tune MusicGen on audio from YouTube videos

## Fine-tuning SDXL

Just give it a URL and a model name on Replicate, and it’ll download the video, take screenshots of every 50 frames, remove near duplicates and very light/dark images, and create a training for you.

```bash
python tune.py
```

## Fine-tuning MusicGen

With `--audio`, it’ll download just the audio, convert it to mp3 and create a training for you.

```bash
python tune.py --audio
```

## Setup

Clone this repo, and setup and activate a virtualenv:

```bash
python3 -m pip install virtualenv
python3 -m virtualenv venv
source venv/bin/activate
```

Then, install the dependencies:
`pip install -r requirements.txt`

Make a [Replicate](https://replicate.com) account and set your token:

`export REPLICATE_API_TOKEN=`

## Run it!

```bash
python tune.py
```