mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-02-19 11:56:43 -05:00
cc4fa96831
The PyTorch decomposition for the op `aten.upsample_bilinear2d.vec` is merged in the upstream repo and hence removed from this file.
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