root@ax650:~/samples/sd1.5-lcm.axera# python3 run_txt2img_axe_infer.py
[INFO] Available providers: ['AxEngineExecutionProvider']
/root/site-packages/local/lib/python3.10/dist-packages/torchvision/datapoints/__init__.py:12: UserWarning: The torchvision.datapoints and torchvision.transforms.v2 namespaces are still Beta. While we do not expect major breaking changes, some APIs may still change according to user feedback. Please submit any feedback you may have in this issue: https://github.com/pytorch/vision/issues/6753, and you can also check out https://github.com/pytorch/vision/issues/7319 to learn more about the APIs that we suspect might involve future changes. You can silence this warning by calling torchvision.disable_beta_transforms_warning().
warnings.warn(_BETA_TRANSFORMS_WARNING)
/root/site-packages/local/lib/python3.10/dist-packages/torchvision/transforms/v2/__init__.py:54: UserWarning: The torchvision.datapoints and torchvision.transforms.v2 namespaces are still Beta. While we do not expect major breaking changes, some APIs may still change according to user feedback. Please submit any feedback you may have in this issue: https://github.com/pytorch/vision/issues/6753, and you can also check out https://github.com/pytorch/vision/issues/7319 to learn more about the APIs that we suspect might involve future changes. You can silence this warning by calling torchvision.disable_beta_transforms_warning().
warnings.warn(_BETA_TRANSFORMS_WARNING)
prompt: Self-portrait oil painting, a beautiful cyborg with golden hair, 8k
text_tokenizer: ./models/tokenizer
text_encoder: ./models/text_encoder
unet_model: ./models/unet.axmodel
vae_decoder_model: ./models/vae_decoder.axmodel
time_input: ./models/time_input_txt2img.npy
save_dir: ./txt2img_output_axe.png
text encoder take 1683.2ms
[INFO] Using provider: AxEngineExecutionProvider
[INFO] Chip type: ChipType.MC50
[INFO] VNPU type: VNPUType.DISABLED
[INFO] Engine version: 2.7.2a
[INFO] Model type: 2 (triple core)
[INFO] Compiler version: 3.3 972f38ca
[INFO] Using provider: AxEngineExecutionProvider
[INFO] Model type: 2 (triple core)
[INFO] Compiler version: 3.3 972f38ca
load models take 4731.6ms
unet once take 449.8ms
unet once take 433.0ms
unet once take 433.0ms
unet once take 433.1ms
unet loop take 1760.8ms
vae inference take 907.8ms
save image take 393.5ms
Exception ignored in: <function AXEngineSession.__del__ at 0xffff513cdc60>
Traceback (most recent call last):
File "/usr/local/lib/python3.10/dist-packages/axengine/_axe.py", line 261, in __del__
File "/usr/local/lib/python3.10/dist-packages/axengine/_axe.py", line 322, in _unload
AttributeError: 'NoneType' object has no attribute 'AX_ENGINE_DestroyHandle'
Exception ignored in: <function AXEngineSession.__del__ at 0xffff513cdc60>
Traceback (most recent call last):
File "/usr/local/lib/python3.10/dist-packages/axengine/_axe.py", line 261, in __del__
File "/usr/local/lib/python3.10/dist-packages/axengine/_axe.py", line 322, in _unload
AttributeError: 'NoneType' object has no attribute 'AX_ENGINE_DestroyHandle'
