AMD-SHARK-Studio/tank/pytorch/v_diffusion_pytorch

v-diffusion model

Installation

Installation (Linux)

Activate shark.venv Virtual Environment

source shark.venv/bin/activate

# Some older pip installs may not be able to handle the recent PyTorch deps
python -m pip install --upgrade pip

Install v-diffusion model and its dependencies

cd tank/pytorch/v_diffusion/
Run the script setup_v_diffusion_pytorch.sh

Run v-diffusion-pytorch model

./v-diffusion-pytorch/cfg_sample.py "New York City, oil on canvas":5 -n 5 -bs 5

Run the v-diffusion model via torch-mlir

./cfg_sample.py "New York City, oil on canvas":5 -n 1 -bs 1 --steps 2