Checkpoint path for integrations/timm not parsed

[eldarkurtic]

Describe the bug
Checkpoint path returned by zoo_model.download_framework_files(extensions=[".pth"]) is of List[str] type.

sparseml/integrations/timm/train.py

Line 392 in 0c8b1d4

args.initial_checkpoint = zoo_model.download_framework_files(extensions=[".pth"])

create_model(...) from the timm repo expects a str type (only one path)

sparseml/integrations/timm/train.py

Line 397 in 0c8b1d4

model = create_model(

Expected behavior
To have: args.initial_checkpoint = "model.pth" instead of args.initial_checkpoint = ["model.pth"].

To Reproduce
Exact steps to reproduce the behavior: use a local recipe and a SparseZoo checkpoint.

python integrations/timm/train.py \
  /PATH/TO/DATASET/imagenet/ \
  --sparseml-recipe /PATH/TO/RECIPE/recipe.yaml \
  --initial-checkpoint zoo:model/stub/path \
  --dataset imagenet \
  --batch-size 64 \
  --remode pixel --reprob 0.6 --smoothing 0.1 \
  --output models/optimized \
  --model resnet50 \
  --workers 8 \

Errors
stat: path should be string, bytes, os.PathLike or integer, not list

[markurtz]

Fixed with the PR that was merged in, not sure why this didn't close out once that went in.