From b67e986ec30a32cb4c192bdba4af3124d0a3911b Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Fri, 15 Sep 2023 11:53:14 -0400
Subject: [PATCH] fix loading hypernetwork

---
 javascript/extraNetworks.js           | 5 ++++-
 modules/hypernetworks/hypernetwork.py | 6 +++---
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/javascript/extraNetworks.js b/javascript/extraNetworks.js
index ce2e8cf03..e48919ef6 100644
--- a/javascript/extraNetworks.js
+++ b/javascript/extraNetworks.js
@@ -187,7 +187,10 @@ function setupExtraNetworksForTab(tabname) {
 
   const intersectionObserver = new IntersectionObserver((entries) => {
     if (!en) return;
-    for (const el of Array.from(gradioApp().querySelectorAll('.extra-networks-page'))) el.style.height = `${window.opts.extra_networks_height}vh`;
+    for (const el of Array.from(gradioApp().querySelectorAll('.extra-networks-page'))) {
+      el.style.height = `${window.opts.extra_networks_height}vh`;
+      el.parentElement.style.width = '-webkit-fill-available';
+    }
     if (entries[0].intersectionRatio > 0) {
       if (window.opts.extra_networks_card_cover === 'cover') {
         en.style.transition = '';
diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py
index d9cc95e86..7b319b1b7 100644
--- a/modules/hypernetworks/hypernetwork.py
+++ b/modules/hypernetworks/hypernetwork.py
@@ -221,10 +221,10 @@ class Hypernetwork:
             torch.save(optimizer_saved_dict, f"{filename}.optim")
 
     def load(self, filename):
-        self.filename = filename
+        self.filename = filename if os.path.exists(filename) else os.path.join(shared.opts.hypernetwork_dir, filename)
         if self.name is None:
-            self.name = os.path.splitext(os.path.basename(filename))[0]
-        with progress.open(filename, 'rb', description=f'Loading hypernetwork: [cyan]{filename}', auto_refresh=True, console=shared.console) as f:
+            self.name = os.path.splitext(os.path.basename(self.filename))[0]
+        with progress.open(self.filename, 'rb', description=f'Loading hypernetwork: [cyan]{self.filename}', auto_refresh=True, console=shared.console) as f:
             state_dict = torch.load(f, map_location='cpu')
         self.layer_structure = state_dict.get('layer_structure', [1, 2, 1])
         self.optional_info = state_dict.get('optional_info', None)