Decode generation prompts inside WSL
This commit is contained in:
parent
974656a13b
commit
2f6a5bd2ce
1 changed files with 2 additions and 1 deletions
|
|
@ -267,6 +267,7 @@ async function generateMotion(request, response) {
|
|||
const duration = Number(body.durationSeconds || body.duration || 4);
|
||||
const diffusionSteps = Math.max(1, Number(body.diffusionSteps || body.diffusion_steps || 20));
|
||||
const seed = body.seed === undefined ? null : Number(body.seed);
|
||||
const promptBase64 = Buffer.from(prompt, "utf8").toString("base64");
|
||||
const linuxOutput = `/mnt/e/Projects/Vili/.vili/artifacts/${jobId}`;
|
||||
const outputStem = "/outputs/motion";
|
||||
const optionalArgs = [];
|
||||
|
|
@ -276,6 +277,7 @@ async function generateMotion(request, response) {
|
|||
"set -e",
|
||||
"service docker start >/dev/null 2>&1 || true",
|
||||
dockerHuggingFaceEnv,
|
||||
`export VILI_PROMPT="$(printf '%s' '${promptBase64}' | base64 -d)"`,
|
||||
`mkdir -p ${linuxOutput}`,
|
||||
[
|
||||
"docker run --rm --runtime=nvidia --gpus all",
|
||||
|
|
@ -294,7 +296,6 @@ async function generateMotion(request, response) {
|
|||
|
||||
const result = await wsl(command, {
|
||||
timeout: Number(body.timeoutMs || 1000 * 60 * 30),
|
||||
env: { VILI_PROMPT: prompt },
|
||||
});
|
||||
|
||||
const job = {
|
||||
|
|
|
|||
Loading…
Reference in a new issue