From 7a8c52cb38c1806f6ef144e2103eff84b5d5dfcc Mon Sep 17 00:00:00 2001
From: Matthias Reso <13337103+mreso@users.noreply.github.com>
Date: Wed, 29 May 2024 21:27:52 +0000
Subject: [PATCH] Remove pkg_resources.packaging
---
src/llama_recipes/utils/train_utils.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/src/llama_recipes/utils/train_utils.py b/src/llama_recipes/utils/train_utils.py
index f62da5bc..ab85e3f4 100644
--- a/src/llama_recipes/utils/train_utils.py
+++ b/src/llama_recipes/utils/train_utils.py
@@ -6,7 +6,6 @@ import time
import yaml
from contextlib import nullcontext
from pathlib import Path
-from pkg_resources import packaging
from datetime import datetime
import contextlib
@@ -474,7 +473,7 @@ def get_policies(cfg, rank):
verify_bfloat_support = ((
torch.version.cuda
and torch.cuda.is_bf16_supported()
- and packaging.version.parse(torch.version.cuda).release >= (11, 0)
+ and torch.version.cuda >= "11.0"
and dist.is_nccl_available()
and nccl.version() >= (2, 10)
) or
--
GitLab