From 54d186671432c9941e2ded141c7c4f472a1ba7d1 Mon Sep 17 00:00:00 2001
From: Saurav Maheshkar <sauravvmaheshkar@gmail.com>
Date: Thu, 1 Aug 2024 09:35:15 +0100
Subject: [PATCH] docs: add features section on README

---
 .github/README.md | 11 ++++++++++-
 sam2/__init__.py  |  4 ++--
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/.github/README.md b/.github/README.md
index 3eefa48c3..a4cf8350c 100644
--- a/.github/README.md
+++ b/.github/README.md
@@ -13,9 +13,18 @@ After downloading the official weights, you can use the `load_model()` helper me
 ```python
 from sam2 import load_model
 
-model = load_model(variant="tiny", ckpt_path="artifacts/sam2_hiera_tiny.pt", device="cpu")
+model = load_model(
+    variant="tiny",
+    ckpt_path="artifacts/sam2_hiera_tiny.pt",
+    device="cpu"
+)
 ```
 
+## Features 🚀
+
+* CPU compatible
+* ships with config files
+
 ## Citation
 
 ```bibtex
diff --git a/sam2/__init__.py b/sam2/__init__.py
index 5ba2d978a..d05adb1c4 100644
--- a/sam2/__init__.py
+++ b/sam2/__init__.py
@@ -4,8 +4,8 @@
 # This source code is licensed under the license found in the
 # LICENSE file in the root directory of this source tree.
 
-from hydra import initialize_config_module
+from hydra import initialize
 
 from .build_sam import load_model
 
-initialize_config_module("sam2.configs", version_base="1.2")
+initialize("configs", version_base="1.2")