chore: add vibe transfer example and update README.md

Author: k3scat

README.md

@@ -202,6 +202,38 @@ for (const image of images) {
 }
 ```
 
+### Vibe Transfer (V4 model only)
+
+All V4 models support vibe transfer, which allows you to transfer the artistic style and mood from reference images to your generated content.
+
+```javascript
+import { NovelAI, Model, Resolution, parseImage } from "nekoai-js";
+
+// Initialize client
+const client = new NovelAI({
+  token: "your_access_token",
+});
+
+// Parse reference image for vibe transfer
+const vibeReference = await parseImage("./input/reference_style.png");
+
+// Generate image with vibe transfer
+const images = await client.generateImage({
+  prompt: "1girl, cute, detailed",
+  model: Model.V4_5,
+  resPreset: Resolution.NORMAL_PORTRAIT,
+  reference_image_multiple: [vibeReference.base64], // Reference image will be converted vibe token (process in the background)
+  reference_information_extracted_multiple: [0.7], // Extraction strength (0.0-1.0)
+  steps: 30,
+  seed: 3417044607,
+});
+
+// Process the resulting images
+for (const image of images) {
+  await image.save("./output");
+}
+```
+
 ### Image to Image
 
 To perform `img2img` action, set `action` parameter to `Action.IMG2IMG`, and provide a source image. Use the `parseImage` utility to handle multiple image formats seamlessly.

examples/model-v4-5-vibe-transfer.js

@@ -0,0 +1,65 @@
+const {
+  NovelAI,
+  Model,
+  Resolution,
+  Sampler,
+  parseImage,
+} = require("nekoai-js");
+const Stream = require("stream");
+require("dotenv").config();
+
+// Test function for V4 model generation
+async function testModelV45Full() {
+  const client = new NovelAI({
+    token: process.env.NOVELAI_TOKEN,
+    verbose: true,
+  });
+
+  console.log("Testing Model V4.5 Vibe Transfer generation...");
+
+  try {
+    const vibe_reference = await parseImage(
+      "./examples/input/vibe_reference.png",
+    );
+
+    const images = await client.generateImage(
+      {
+        prompt: "1girl, cute",
+        negative_prompt: "1234",
+        model: Model.V4_5,
+        resPreset: Resolution.NORMAL_PORTRAIT,
+        seed: 3417044607,
+        steps: 30,
+        scale: 5,
+        sampler: Sampler.EULER_ANC,
+        reference_image_multiple: [vibe_reference.base64], // the reference image will be re-encoded as vibe token
+        reference_information_extracted_multiple: [0.7],
+        ucPreset: 3,
+      },
+      false,
+      true,
+    ); // Set verbose to true to see Anlas cost
+
+    if (images.length > 0) {
+      console.log(`Generated ${images.length} image(s)`);
+      // Save the images to the output directory
+      for (const [index, image] of images.entries()) {
+        const path = await image.save(
+          "./examples/output/model-v4-5-full-vibe.png",
+        );
+        console.log(`Saved image ${index + 1} to ${path}`);
+      }
+    } else {
+      console.log("No images were generated");
+    }
+  } catch (error) {
+    console.error("Error generating image:", error);
+  }
+}
+
+// Run the test if this script is executed directly
+if (require.main === module) {
+  testModelV45Full().catch(console.error);
+}
+
+module.exports = { testModelV45Full };