You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
After running the code until the compilation part, the models do not exist. The compilation logs indicate that the process completes without errors, but the expected model file model.pt is missing from the directory sd2_compile_dir_768/unet/.
Steps to Reproduce
Activate the pre-built PyTorch-2.1 environment for Inf2, Trn*:
Run the provided template code from the repository:
python3test3.py
Observe the logs and check for the existence of the model file in the specified directory.
Expected Behavior
The model file model.pt should be present in the directory sd2_compile_dir_768/unet/ after the compilation process completes.
Actual Behavior
The model file model.pt is missing from the directory sd2_compile_dir_768/unet/.
Compilation Logs
2024-05-30T09:32:51Z Running birverifier
2024-05-30T09:32:52Z birverifier finished after 1.166 seconds
2024-05-30T09:32:52Z Running codegen
2024-05-30T09:32:57Z isa_gen finished after 4.293 seconds
2024-05-30T09:32:58Z dma_desc_gen finished after 1.495 seconds
2024-05-30T09:33:01Z debug_info_gen finished after 2.790 seconds
2024-05-30T09:33:02Z codegen finished after 9.213 seconds
2024-05-30T09:33:02Z Running neff_packager
2024-05-30T09:33:29Z neff_packager finished after 27.627 seconds
Error Message
Traceback (most recent call last):
File "/home/ubuntu/test3.py", line 124, in <module>
pipe.unet.unetwrap = torch_neuronx.DataParallel(torch.jit.load(unet_filename), device_ids, set_dynamic_batching=False)
File "/opt/aws_neuronx_venv_pytorch_2_1/lib/python3.10/site-packages/torch/jit/_serialization.py", line 152, in load
raise ValueError(f"The provided filename {f} does not exist") # type: ignore[str-bytes-safe]
ValueError: The provided filename sd2_compile_dir_768/unet/model.pt does not exist
The model file model.pt is missing from the directory sd2_compile_dir_768/unet/.
Hi, I routed the issue to the appropriate team within the org. We will track it and get back to you.
Description
After running the code until the compilation part, the models do not exist. The compilation logs indicate that the process completes without errors, but the expected model file
model.pt
is missing from the directorysd2_compile_dir_768/unet/
.Steps to Reproduce
source /opt/aws_neuronx_venv_pytorch_2_1/bin/activate
Expected Behavior
The model file
model.pt
should be present in the directorysd2_compile_dir_768/unet/
after the compilation process completes.Actual Behavior
The model file
model.pt
is missing from the directorysd2_compile_dir_768/unet/
.Compilation Logs
Error Message
Environment Details
Additional Information
test.py (for compilation)
Screenshots
The text was updated successfully, but these errors were encountered: