Skip to content

Conversation

@mckellcarter
Copy link

Summary

  • Removes dependency on private torchvision functions (_get_confirm_token, _save_response_content)
  • Updates to use torchvision's public API for Google Drive downloads
  • Removes version upper limits for torch and torchvision
  • Adds gdown as required dependency

Fixes

  • Resolves ModuleNotFoundError with torchvision>=0.17
  • Enables compatibility with torchvision 0.20.x+

Testing

Tested with torchvision 0.20.1 - package loads without errors or warnings

🤖 Generated with Claude Code

This update removes dependency on private torchvision functions that were removed in newer versions and modernizes the package to work with current PyTorch ecosystem.

Changes:
- Replace custom Google Drive download implementation with torchvision's public API
- Remove dependency on private functions (_get_confirm_token, _save_response_content)
- Remove version upper limits for torch and torchvision in setup.py
- Add gdown as a required dependency (needed by torchvision>=0.17)

This fixes the ModuleNotFoundError for '_get_confirm_token' that occurs with torchvision>=0.17.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant