wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 21:42:37 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

fix python

Browse Source
This commit is contained in:
pythongosssss 2023-10-20 18:40:40 +01:00
parent 57d1bba5eb
commit e09c8a351f
2 changed files with 57 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -14,3 +14,4 @@ venv/
/web/extensions/*
!/web/extensions/logging.js.example
!/web/extensions/core/
/tests-ui/data/object_info.json

109
tests-ui/setup.js
View File

@ -1,68 +1,71 @@
const { spawn } = require("child_process");
const { resolve } = require("path");
const { writeFile } = require("fs");
const { writeFile, existsSync } = require("fs");
const http = require("http");
async function setup() {
// Wait up to 30s for it to start
let success = false;
let child;
for (let i = 0; i < 30; i++) {
try {
await new Promise((res, rej) => {
http
.get("http://127.0.0.1:8188/object_info", (resp) => {
let data = "";
resp.on("data", (chunk) => {
data += chunk;
});
resp.on("end", () => {
// Modify the response data to add some checkpoints
const objectInfo = JSON.parse(data);
objectInfo.CheckpointLoaderSimple.input.required.ckpt_name[0] = ["model1.safetensors", "model2.ckpt"];
async function setup() {
// Wait up to 30s for it to start
let success = false;
let child;
for (let i = 0; i < 30; i++) {
try {
await new Promise((res, rej) => {
http
.get("http://127.0.0.1:8188/object_info", (resp) => {
let data = "";
resp.on("data", (chunk) => {
data += chunk;
});
resp.on("end", () => {
// Modify the response data to add some checkpoints
const objectInfo = JSON.parse(data);
objectInfo.CheckpointLoaderSimple.input.required.ckpt_name[0] = ["model1.safetensors", "model2.ckpt"];
data = JSON.stringify(objectInfo, undefined, "\t");
data = JSON.stringify(objectInfo, undefined, "\t");
writeFile(
"data/object_info.json",
data,
{
encoding: "utf8",
},
(err) => {
if (err) {
rej(err);
} else {
res();
}
writeFile(
"data/object_info.json",
data,
{
encoding: "utf8",
},
(err) => {
if (err) {
rej(err);
} else {
res();
}
);
res();
});
})
.on("error", rej);
});
success = true;
break;
} catch (error) {
if (i === 0) {
// Start the server on first iteration if it fails to connect
console.log("Starting ComfyUI server...");
}
);
res();
});
})
.on("error", rej);
});
success = true;
break;
} catch (error) {
if (i === 0) {
// Start the server on first iteration if it fails to connect
console.log("Starting ComfyUI server...");
const python = resolve("../../python_embeded/python.exe");
child = spawn(python, ["-s", "ComfyUI/main.py", "--cpu"], { cwd: "../.." });
let python = resolve("../../python_embeded/python.exe");
if (!existsSync(python)) {
python = "python";
}
await new Promise((r) => {
setTimeout(r, 1000);
});
child = spawn(python, ["-s", "ComfyUI/main.py", "--cpu"], { cwd: "../.." });
}
await new Promise((r) => {
setTimeout(r, 1000);
});
}
}
child?.kill();
child?.kill();
if (!success) {
throw new Error("Waiting for server timed out...");
}
}
if (!success) {
throw new Error("Waiting for server timed out...");
}
}
setup();