1
0
Fork 0

load from encoder path, add default param

This commit is contained in:
Sean Sube 2023-02-21 23:16:13 -06:00
parent 2e7de16778
commit 7ad8385c5b
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 6 additions and 2 deletions

View File

@ -1,4 +1,5 @@
from logging import getLogger
from os import path
from typing import Any, Optional, Tuple
import numpy as np
@ -189,10 +190,9 @@ def load_pipeline(
if inversion is not None:
logger.debug("loading text encoder from %s", inversion)
text_encoder = OnnxRuntimeModel.from_pretrained(
inversion,
path.join(inversion, "text_encoder"),
provider=device.ort_provider(),
sess_options=device.sess_options(),
subfolder="text_encoder",
)
pipe = pipeline.from_pretrained(

View File

@ -60,6 +60,10 @@
"max": 1024,
"step": 8
},
"inversion": {
"default": "",
"keys": []
},
"left": {
"default": 0,
"min": 0,