From d508a59eb4151170318409c16cd5feee98a6e6bb Mon Sep 17 00:00:00 2001
From: "sweep-ai[bot]" <128439645+sweep-ai[bot]@users.noreply.github.com>
Date: Fri, 27 Oct 2023 18:15:00 +0000
Subject: [PATCH 1/2] feat: Updated beyond_the_nest/__main__.py

---
 beyond_the_nest/__main__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/beyond_the_nest/__main__.py b/beyond_the_nest/__main__.py
index 2c8aa59..b90e541 100644
--- a/beyond_the_nest/__main__.py
+++ b/beyond_the_nest/__main__.py
@@ -1,4 +1,4 @@
-from beyond_the_nest.beyond_the_nest import generate_text, load_transformers_model
+from beyond_the_nest.beyond_the_nest_models import generate_text, load_falcon_model_generation as load_transformers_model
 
 # Example usage:
 model_name = "tiiuae/falcon-7b-instruct"

From 2985ac53bb3ef1eca9ee8c5c40a5c1cdfb40aac3 Mon Sep 17 00:00:00 2001
From: "sweep-ai[bot]" <128439645+sweep-ai[bot]@users.noreply.github.com>
Date: Fri, 27 Oct 2023 18:16:05 +0000
Subject: [PATCH 2/2] Sandbox run beyond_the_nest/__main__.py

---
 beyond_the_nest/__main__.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/beyond_the_nest/__main__.py b/beyond_the_nest/__main__.py
index b90e541..47798f4 100644
--- a/beyond_the_nest/__main__.py
+++ b/beyond_the_nest/__main__.py
@@ -1,4 +1,7 @@
-from beyond_the_nest.beyond_the_nest_models import generate_text, load_falcon_model_generation as load_transformers_model
+from beyond_the_nest.beyond_the_nest_models import generate_text
+from beyond_the_nest.beyond_the_nest_models import (
+    load_falcon_model_generation as load_transformers_model,
+)
 
 # Example usage:
 model_name = "tiiuae/falcon-7b-instruct"