Skip to content

Commit

Permalink
Create README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Yoinky3000 authored Jun 14, 2024
1 parent 3a42fbd commit abfbf55
Showing 1 changed file with 77 additions and 0 deletions.
77 changes: 77 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,77 @@
<div align="center">

# Docker Image for SD Next

</div>

A docker image to automatically setup everything you needed to run SD Next in a container

## Image Configurations
* Ubuntu 22.04 LTS
* Support CUDA 11.8 ~ 12.5
* Python 3.10
* PyTorch 2.3.1
* SD Next ([84f9caa](https://github.com/vladmandic/automatic/tree/84f9caaffe8eae40f2f597c09faccb59a4eab2b7))
* Jupyter Lab Installed

## Features
* [Original + Diffusers Backend supported](https://github.com/vladmandic/automatic?tab=readme-ov-file#backend-support)
* [Vast majority of model types supported](https://github.com/vladmandic/automatic?tab=readme-ov-file#model-support)
* Plug'n'Play (packages pre-installed)
* Easy management(start, git version, download from links) with jupyter notebook

## Deploy
* ### Runpod
You can deploy this image in [Runpod](https://runpod.io?ref=2v9nfixx) with this [template](https://runpod.io/console/deploy?template=joh7y33050&ref=2v9nfixx)<br>
The template use cu124 version, so please remember to use the pod that support CUDA 12.4
* ### Local
Run the following command in your terminal:
```bash
docker run -d \
--gpus all \
-v /workspace \
-p PORT:3000 \
-p PORT:3001 \
-e JUPYTER_LAB_PASSWORD="abcd1234" \
yoinky3000/sd-next-docker:x.x.x-cuxxx
```

> [!NOTE]
>
> Replace x.x.x with the version of this image you want to use (`:latest` tag will install latest version for CUDA 12.4)
>
> You will need to replace `PORT` with the port number you want the apps to listen to,
> For the details of the ports, please scroll down to the [PORTS](#PORTS) section
>
> You can change the values of JUPYTER_LAB_PASSWORD, or remove it from the command as you like
> [!IMPORTANT]
> Each version of the image will be built for each [CUDA version listed here](#Image-Configurations) specifically
>
> To check which CUDA version of the image is suitable, open the terminal and use `nvidia-smi` to check
> the CUDA version your system has installed, you should see `CUDA Version: XX.Y` in the output (below is an example)
> ![Screenshot 2024-06-13 230443](https://github.com/Yoinky3000/sd-next-docker/assets/65208589/adf662bf-cacb-4a0d-a6be-7bdc396a39b3)
> now you can replace cuxxx with cuXXY


## PORTS
| Internal Port | Apps |
|---------------|-------------------------------|
| 3000 | Jupyter LAB |
| 3001 | SD Next |

### Environment Variables

| Variable | Description | Default |
|----------------------|---------------------------------------------------|---------------------------------------------------|
| JUPYTER_LAB_PASSWORD | This will be used as the password for jupyter lab | None (no password is required to access the lab) |

## Contribution
Contributions are welcome! Create an issue if you have any problem with this image, or create pull request if you want to add new features

## Disclaimer
The authors of this project are not responsible for any content generated using this interface.

## Any support will be highly appreciated
<a href="https://www.buymeacoffee.com/yoinky3000" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/v2/default-yellow.png" alt="Buy Me A Coffee" style="height: 40px !important;width: 170px !important;" ></a>

0 comments on commit abfbf55

Please sign in to comment.