apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

add safetensors dep

This commit is contained in:
Sean Sube 2023-02-09 00:01:48 -06:00
parent beb3411811
commit 36d164a281
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/convert/diffusion_original.py
View File

@ -1607,7 +1607,7 @@ def extract_checkpoint(ctx: ConversionContext, new_model_name: str, checkpoint_f
def convert_diffusion_original(ctx: ConversionContext, model_name: str, tensor_file: str, opset: int, half: bool):
model_path = os.path.join(ctx.model_path, model_name)
torch_name = model_name.replace("-onnx-", "-torch-")
torch_name = model_name.replace("onnx", "torch")
torch_path = os.path.join(ctx.model_path, torch_name)
working_name = os.path.join(ctx.model_path, torch_name, "working")
logger.info("Converting original Diffusers checkpoint %s: %s -> %s", model_name, tensor_file, model_path)

3
api/requirements.txt
View File

@ -25,4 +25,5 @@ pyyaml
# TODO: get rid of these
omegaconf
pydantic
pydantic
safetensors