fixed slice node output type

This commit is contained in:
SayanoAI
2024-08-07 15:28:27 -04:00
parent c877b8c770
commit 5b12b19bc4
2 changed files with 6 additions and 4 deletions
+1 -1
View File
@@ -43,7 +43,7 @@ if %errorlevel% == 0 (
rem install CUDA torch
CALL pip install torch==2.0.1+cu118 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
CALL python main.py --disable-smart-memory
CALL python main.py --disable-smart-memory --preview-method=taesd --force-fp16 --deterministic --use-pytorch-cross-attention
) else (
echo Failed to activate environment...
)
+5 -3
View File
@@ -303,9 +303,11 @@ app.registerExtension({
break;
case "SliceNode":
chainCallback(nodeType.prototype, "onConnectInput", function (_, inputs) {
this.outputs[0].name = inputs;
this.outputs[0].type = inputs;
chainCallback(nodeType.prototype, "onConnectInput", function (slot, inputs) {
if (slot===0){
this.outputs[0].name = inputs;
this.outputs[0].type = inputs;
}
})
break;