1
0
Fork 0

add test script, screenshot to readme

This commit is contained in:
Sean Sube 2023-01-05 09:44:01 -06:00
parent 47a64f5a36
commit 6c932e2ee8
6 changed files with 27 additions and 10 deletions

View File

@ -1,7 +1,11 @@
# ONNX Web UI # ONNX Web UI
This is a rudimentary web UI for ONNX models, providing a way to run GPU-accelerated models on Windows and AMD with a This is a rudimentary web UI for ONNX models, providing a way to run GPU-accelerated models on Windows and even AMD
remote web interface. with a remote web interface.
This is still fairly early and instructions are a little rough, but it works on my machine.
![txt2img with example astronaut prompt and image](./docs/readme-preview.png)
Based on work by: Based on work by:
@ -17,7 +21,7 @@ Based on work by:
- [Setup](#setup) - [Setup](#setup)
- [Install Git and Python](#install-git-and-python) - [Install Git and Python](#install-git-and-python)
- [Create a Virtual Environment](#create-a-virtual-environment) - [Create a Virtual Environment](#create-a-virtual-environment)
- [Install pip packages](#install-pip-packages) - [Install AI packages](#install-ai-packages)
- [Install ORT Nightly package](#install-ort-nightly-package) - [Install ORT Nightly package](#install-ort-nightly-package)
- [Download and Convert Models](#download-and-convert-models) - [Download and Convert Models](#download-and-convert-models)
- [Usage](#usage) - [Usage](#usage)
@ -28,10 +32,15 @@ Based on work by:
## Setup ## Setup
This is a very similar process to what harishanand95 and averad's gists recommend, split up into a few larger steps: This is a very similar process to what [harishanand95](https://gist.github.com/harishanand95/75f4515e6187a6aa3261af6ac6f61269)
and [averad's](https://gist.github.com/averad/256c507baa3dcc9464203dc14610d674) gists recommend, split up into a few
steps:
1. Create a virtual environment 1. Install Git and Python, if you have not already
2. Install pip package 2. Create a virtual environment
3. Install AI packages
4. Install ORT Nightly package
5. Download and convert models
### Install Git and Python ### Install Git and Python
@ -68,7 +77,7 @@ Every time you start using ONNX web, activate the virtual environment:
> .\onnx_env\Scripts\Activate.bat > .\onnx_env\Scripts\Activate.bat
``` ```
### Install pip packages ### Install AI packages
Update pip itself: Update pip itself:

11
api/test-setup.py Normal file
View File

@ -0,0 +1,11 @@
from diffusers import OnnxStableDiffusionPipeline
cfg=8
steps=22
height=512
width=512
prompt = "an astronaut eating a hamburger"
pipe = OnnxStableDiffusionPipeline.from_pretrained("../../stable_diffusion_onnx", provider="DmlExecutionProvider", safety_checker=None)
image = pipe(prompt, height, width, num_inference_steps=steps, guidance_scale=cfg).images[0]
image.save("test.png")

0
docs/api.md Normal file
View File

0
docs/getting-started.md Normal file
View File

BIN
docs/readme-preview.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 346 KiB

View File

@ -39,9 +39,6 @@ export function Txt2Img(props: Txt2ImgProps) {
return <Box> return <Box>
<Stack spacing={2}> <Stack spacing={2}>
<Box>
txt2img mode
</Box>
<Select <Select
value={scheduler} value={scheduler}
label="Scheduler" label="Scheduler"