diff --git a/deepseek_vl/__init__.py b/deepseek_vl/__init__.py index 8cb7640..8f00116 100644 --- a/deepseek_vl/__init__.py +++ b/deepseek_vl/__init__.py @@ -16,3 +16,14 @@ # COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER # IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN # CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. + + +# check if python version is above 3.10 +import sys +if sys.version_info >=(3, 10): + print("Python version is above 3.10, patching the collections module.") + # Monkey patch collections + import collections + import collections.abc + for type_name in collections.abc.__all__: + setattr(collections, type_name, getattr(collections.abc, type_name))