diff --git a/docker/smoke_test.py b/docker/smoke_test.py index 2c1a6442f1..a2526421ae 100644 --- a/docker/smoke_test.py +++ b/docker/smoke_test.py @@ -69,6 +69,7 @@ def check_imports() -> None: # smoke-tests both arches. try: import xformers + print(f"xformers {xformers.__version__}") except ImportError: print("xformers (missing -- expected on arm64 [huggingface] extras)")