-
Notifications
You must be signed in to change notification settings - Fork 1
/
img2img.js
61 lines (57 loc) · 1.75 KB
/
img2img.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
/** @format */
const { NovitaSDK } = require("novita-sdk");
const path = require("path");
const { convertImageToBase64 } = require("./utils.js");
const novitaClient = new NovitaSDK(process.env.NOVITA_API_KEY);
async function img2img(onFinish) {
const baseImg = await convertImageToBase64(path.join(__dirname, "test.png"));
const params = {
init_images: [baseImg],
model_name: "majicmixRealistic_v7_134792.safetensors",
prompt: "1girl,sweater,white background",
negative_prompt: "(worst quality:2),(low quality:2),(normal quality:2),lowres,watermark,",
width: 512,
height: 768,
sampler_name: "Euler a",
cfg_scale: 7,
steps: 20,
batch_size: 1,
n_iter: 1,
seed: -1,
};
novitaClient
.img2img(params)
.then((res) => {
if (res && res.task_id) {
const timer = setInterval(() => {
novitaClient
.progress({
task_id: res.task_id,
})
.then((progressRes) => {
if (progressRes.status === 2) {
console.log("finished!", progressRes.imgs);
clearInterval(timer);
onFinish(progressRes.imgs);
}
if (progressRes.status === 3 || progressRes.status === 4) {
console.warn("failed!", progressRes.failed_reason);
clearInterval(timer);
}
if (progressRes.status === 1) {
console.log("progress", progressRes.current_images);
}
})
.catch((err) => {
console.error("progress error:", err);
});
}, 1000);
}
})
.catch((err) => {
console.error("txt2Img error:", err);
});
}
img2img((imgs) => {
console.log(imgs);
});