[feat]: General diffusers adapter backend to run diffusion models (#2724)
Signed-off-by: Huang, Zeyu <11222265+fhfuih@users.noreply.github.com>
This commit is contained in:
1 parent
a664e40716
commit
d8cc7a0409
13 files changed
+936
-24
No files matched your search
@@ -1319,6 +1319,12 @@ class AsyncOmniEngine:
|
||||
if "dtype" in normalized_kwargs:
|
||||
stage_engine_args["dtype"] = normalized_kwargs["dtype"]
|
||||
|
||||
# New split fields for diffusers adapter kwargs.
|
||||
if kwargs.get("diffusers_load_kwargs") is not None:
|
||||
stage_engine_args["diffusers_load_kwargs"] = kwargs["diffusers_load_kwargs"]
|
||||
if kwargs.get("diffusers_call_kwargs") is not None:
|
||||
stage_engine_args["diffusers_call_kwargs"] = kwargs["diffusers_call_kwargs"]
|
||||
|
||||
default_stage_cfg = [
|
||||
{
|
||||
"stage_id": 0,
|
||||
|
||||
Reference in new issue
Block a user