-
Notifications
You must be signed in to change notification settings - Fork 11
Description
Thanks for you wonderful work! When I tried to FT the SD model and ran the train_SDFT.py, I met the issue below:
I wonder if the version of PIL is right?
Total trainable parameters: 3.19488M of 869.120004M.
Map: 0%| | 0/30 [00:00<?, ? examples/s]
Traceback (most recent call last):
File "/root/autodl-tmp/Pano2Room/train_SDFT.py", line 801, in
main(args)
File "/root/autodl-tmp/Pano2Room/train_SDFT.py", line 597, in main
train_dataset = train_dataset.map(
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/arrow_dataset.py", line 602, in wrapper
out: Union["Dataset", "DatasetDict"] = func(self, *args, **kwargs)
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/arrow_dataset.py", line 567, in wrapper
out: Union["Dataset", "DatasetDict"] = func(self, *args, **kwargs)
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/arrow_dataset.py", line 3167, in map
for rank, done, content in Dataset._map_single(**dataset_kwargs):
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/arrow_dataset.py", line 3558, in _map_single
batch = apply_function_on_filtered_inputs(
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/arrow_dataset.py", line 3411, in apply_function_on_filtered_inputs
inputs = format_table(
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/formatting/formatting.py", line 639, in format_table
return formatter(pa_table, query_type=query_type)
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/formatting/formatting.py", line 407, in call
return self.format_batch(pa_table)
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/formatting/formatting.py", line 521, in format_batch
batch = self.python_features_decoder.decode_batch(batch)
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/formatting/formatting.py", line 228, in decode_batch
return self.features.decode_batch(batch) if self.features else batch
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/features/features.py", line 2081, in decode_batch
[
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/features/features.py", line 2082, in
decode_nested_example(self[column_name], value, token_per_repo_id=token_per_repo_id)
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/features/features.py", line 1400, in decode_nested_example
return schema.decode_example(obj, token_per_repo_id=token_per_repo_id)
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/datasets/features/image.py", line 189, in decode_example
if image.getexif().get(PIL.Image.ExifTags.Base.Orientation) is not None:
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/PIL/Image.py", line 108, in getattr
raise AttributeError(f"module '{name}' has no attribute '{name}'")
AttributeError: module 'PIL.Image' has no attribute 'ExifTags'
Traceback (most recent call last):
File "/root/miniconda3/envs/P2R/bin/accelerate", line 8, in
sys.exit(main())
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/accelerate/commands/accelerate_cli.py", line 48, in main
args.func(args)
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/accelerate/commands/launch.py", line 1106, in launch_command
simple_launcher(args)
File "/root/miniconda3/envs/P2R/lib/python3.9/site-packages/accelerate/commands/launch.py", line 704, in simple_launcher
raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)
subprocess.CalledProcessError: Command '['/root/miniconda3/envs/P2R/bin/python', 'train_SDFT.py']' returned non-zero exit status 1.