search

AUTOMATIC1111 / stable-diffusion-webui-tensorrt

MIT License
311 stars 20 forks source link

ImportError: cannot import name 'sd_unet' from 'modules' (unknown location) #2

Closed Cubey42 closed 1 year ago

Cubey42 commented 1 year ago

Error loading script: trt.py Traceback (most recent call last): File "H:\NovelAI\stable-diffusion-webui\modules\scripts.py", line 263, in load_scripts script_module = script_loading.load_module(scriptfile.path) File "H:\NovelAI\stable-diffusion-webui\modules\script_loading.py", line 10, in load_module module_spec.loader.exec_module(module) File "", line 883, in exec_module File "", line 241, in _call_with_frames_removed File "H:\NovelAI\stable-diffusion-webui\extensions\stable-diffusion-webui-tensorrt\scripts\trt.py", line 6, in from modules import script_callbacks, sd_unet, devices, shared, paths_internal ImportError: cannot import name 'sd_unet' from 'modules' (unknown location)

kilkal-kol commented 1 year ago

Had the same issue. Fixed by using dev branch. "Adds the ability to convert loaded model's Unet module into TensortRT. Requires version least after commit 339b5315 (currently, it's the dev branch after 2023-05-27). Only tested to work on Windows."

Cubey42 commented 1 year ago

thank you.

Qxzz2 commented 1 year ago

Had the same issue. Fixed by using dev branch. "Adds the ability to convert loaded model's Unet module into TensortRT. Requires version least after commit 339b5315 (currently, it's the dev branch after 2023-05-27). Only tested to work on Windows."

Any chance you could elaborate on how to go about implementing this? Also having this issue, but can't resolve.

hiyata commented 1 year ago

Just went through this, open up cmd in your stable-diffusion-webui folder and type "git checkout dev". Then type "webui-user.bat" and you should be able to follow the instructions. Make sure the TensorRT folder is being used, the one inside TensorRT-8.6.1.6..etc.