search

siliconflow / onediff

OneDiff: An out-of-the-box acceleration library for diffusion models.
https://github.com/siliconflow/onediff/wiki
Apache License 2.0
1.7k stars 104 forks source link

No module named 'onediff.infer_compiler.register' #884

Closed Ralphhtt closed 6 months ago

Ralphhtt commented 6 months ago

Describe the bug

Failed to load onediff_sd_webui_extentions *** Error loading script: onediff.py ImportError: cannot import name 'register' from 'onediff.infer_compiler'

Your environment

Ubuntu 22.04.4 LTS

OneDiff git commit id

d8a6a90

OneFlow version info

version: 0.9.1.dev20240515+cu118 git_commit: ec7b682 cmake_build_type: Release rdma: True mlir: True enterprise: False

How To Reproduce

cd onediff python3 -m pip install -e . python >>> from onediff.infer_compiler import oneflow_compile, register

The complete error message

Traceback (most recent call last): File "", line 1, in ImportError: cannot import name 'register' from 'onediff.infer_compiler' (/workspace/onediff/src/onediff/infer_compiler/init.py)

Additional context

Add any other context about the problem here.

strint commented 6 months ago

got it, will fix it shortly

strint commented 6 months ago

@Ralphhtt Fixed in https://github.com/siliconflow/onediff/pull/885 , please update to the main branch.

Ralphhtt commented 6 months ago

It works, thx!