From c0d62ac97efbcd19de85d8e0e834ff232440d981 Mon Sep 17 00:00:00 2001
From: Dhruv Nair <dhruv.nair@gmail.com>
Date: Sat, 30 Dec 2023 11:39:00 +0000
Subject: [PATCH] update

---
 src/diffusers/loaders/single_file.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/diffusers/loaders/single_file.py b/src/diffusers/loaders/single_file.py
index fd5a955316..002f867653 100644
--- a/src/diffusers/loaders/single_file.py
+++ b/src/diffusers/loaders/single_file.py
@@ -148,7 +148,8 @@ def build_component(
     **kwargs,
 ):
     if component_name in kwargs:
-        return kwargs.pop(component_name, None)
+        component = kwargs.pop(component_name, None)
+        return {component_name: component}
 
     if component_name in pipeline_components:
         return {}