Fix docstring & Add missing safetensors requirement#4

perkyfever wants to merge 1 commit intoBlinorot:example/image-classificationBlinorot/pytorch_project_template:example/image-classificationfrom

perkyfever:example/image-classificationperkyfever/pytorch_project_template:example/image-classificationCopy head branch name to clipboard