search

ludwig-ai / ludwig

Low-code framework for building custom LLMs, neural networks, and other AI models
http://ludwig.ai
Apache License 2.0
11.22k stars 1.19k forks source link

[CI] Upgrade torch on CIs to 2.0.0, 2.1.1, and nightly #3785

Closed justinxzhao closed 1 year ago

github-actions[bot] commented 1 year ago

Unit Test Results

  6 files  ±0    6 suites  ±0   14m 30s :stopwatch: - 1m 16s 12 tests ±0    9 :heavy_check_mark: ±0    3 :zzz: ±0  0 :x: ±0  60 runs  ±0  42 :heavy_check_mark: ±0  18 :zzz: ±0  0 :x: ±0 

Results for commit 1aff6804. ± Comparison against base commit 8c47c3cb.

:recycle: This comment has been updated with latest results.

arnavgarg1 commented 1 year ago

LGTM, but we should also update the torch version of the docker images: https://github.com/ludwig-ai/ludwig/tree/master/docker

justinxzhao commented 1 year ago

LGTM, but we should also update the torch version of the docker images: https://github.com/ludwig-ai/ludwig/tree/master/docker

Good call out. Instead of upgrading the CI to torch==2.1.0, pytorch==2.0.0 seems to fix the CI and is already consistent with our docker images.

arnavgarg1 commented 1 year ago

That makes sense and sounds good!