text2vec==1.2.9 和 torch==2.0.0 兼容性问题：发生异常: ImportError (note: full exception trace is shown but execution is paused at: <module>) cannot import name 'LRScheduler' from 'torch.optim.lr_scheduler'

[joy1990]

GitHub Issue 模板

标题：
text2vec==1.2.9 和 torch==2.0.0 兼容性问题：发生异常: ImportError (note: full exception trace is shown but execution is paused at: )
cannot import name 'LRScheduler' from 'torch.optim.lr_scheduler'

描述：
在使用 text2vec==1.2.9 和 torch==2.0.0 时，遇到导入错误 ImportError: cannot import name 'LRScheduler' from 'torch.optim.lr_scheduler'。

详细信息：

1. 环境信息：

   • Python 版本：3.10.12
   • torch 版本：2.0.0
   • text2vec 版本：1.2.9

2. 错误信息：

   ImportError: cannot import name 'LRScheduler' from 'torch.optim.lr_scheduler'
   

3. 复现步骤：

   • 安装依赖：

     pip install torch==2.0.0
     pip install text2vec==1.2.9

   • 运行代码：

     from text2vec import cos_sim, SentenceModel
     
     # 加载预训练的 Sentence-BERT 模型
     model = SentenceModel('shibing624/text2vec-base-chinese')
     
     # 输入文本
     text1 = "酒店非常干净"
     text2 = "这家酒店的房间很整洁"
     
     # 将文本转换为向量
     vec1 = model.encode([text1])[0]
     vec2 = model.encode([text2])[0]
     
     # 使用 torch 进行余弦相似度计算
     similarity = cos_sim(vec1, vec2)
     print(f"余弦相似度: {similarity.item()}")

4. 预期行为：

   • 成功加载 SentenceModel 并计算文本相似度。

5. 实际行为：

   • 报错 ImportError: cannot import name 'LRScheduler' from 'torch.optim.lr_scheduler'

附加信息：

• 是否尝试过其他版本的 torch 或 text2vec？如果尝试过，请描述结果。
• 是否有其他库依赖冲突？如果有，请提供相关信息。

[joy1990]

原来被这哥们改了

[shibing624]

建议升级torch版本到2.4.0及以上。