add requirements file and test script
This commit is contained in:
parent
6c932e2ee8
commit
ccbce9abe8
|
@ -0,0 +1,10 @@
|
|||
### AI packages ###
|
||||
diffusers
|
||||
numpy
|
||||
onnx
|
||||
onnxruntime
|
||||
transformers
|
||||
|
||||
### Server packages ###
|
||||
flask
|
||||
stringcase
|
|
@ -1,11 +1,11 @@
|
|||
from diffusers import OnnxStableDiffusionPipeline
|
||||
|
||||
cfg=8
|
||||
steps=22
|
||||
height=512
|
||||
width=512
|
||||
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")
|
||||
image.save("../test.png")
|
Loading…
Reference in New Issue