Create README.md

README.md

@@ -0,0 +1,15 @@
+StableDiffusion 2.1 converted to fp16 ONNX model to run GPU accelerated NodeJS code https://github.com/dakenf/stable-diffusion-nodejs
+
+```javascript
+import * as tf from "@tensorflow/tfjs-node"
+import { StableDiffusionPipeline } from 'stable-diffusion-nodejs'
+
+const pipe = await StableDiffusionPipeline.fromPretrained(
+  'directml', // can be 'cuda' on linux or 'cpu' on mac os
+  'aislamov/stable-diffusion-2-1-onnx', // relative path or huggingface repo with onnx model
+)
+
+const image = await pipe.run("A photo of a cat", undefined, 1, 9, 30)
+const png = await tf.node.encodePng(image[0])
+fs.writeFileSync("output.png", png);
+```