__pycache__ /venv .vscode *.ckpt *.safetensors *.pth *.pt manual_testing.py