github/AMD-SHARK-Studio

mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-02-19 11:56:43 -05:00

Files

Vivek Khandelwal 0e2485d85b Add v-diffusion model

2022-07-20 21:36:42 +05:30

619 B

Raw Permalink Blame History

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

./setup_diffusion.sh

Run v-diffusion-pytorch model

./v-diffusion-pytorch/cfg_sample.py "the rise of consciousness":5 -n 5 -bs 5 --seed 0

Compile v-diffusion model via torch-mlir

python v_diffusion.py 2> v_diffusion_ir.mlir