update space

app.py

@@ -257,19 +257,20 @@ def main():
         elif 'cyber' in instruction:
             e_type = 'cyber'
 
-        model = load_model('text300M', device=device)
-        with torch.no_grad():
-            new_proj = nn.Linear(1024 * 2, 1024, device=device, dtype=model.wrapped.input_proj.weight.dtype)
-            new_proj.weight = nn.Parameter(torch.zeros_like(new_proj.weight))
-            new_proj.weight[:, :1024].copy_(model.wrapped.input_proj.weight)  #
-            new_proj.bias = nn.Parameter(torch.zeros_like(new_proj.bias))
-            new_proj.bias[:1024].copy_(model.wrapped.input_proj.bias)
-            model.wrapped.input_proj = new_proj
-
-        ckp = torch.load(
-            hf_hub_download(repo_id='silentchen/Shap_Editor', subfolder='single', filename='{}.pt'.format(e_type)),
-            map_location='cpu')
-        model.load_state_dict(ckp['model'])
+        model = models[e_type]
+        # model = load_model('text300M', device=device)
+        # with torch.no_grad():
+        #     new_proj = nn.Linear(1024 * 2, 1024, device=device, dtype=model.wrapped.input_proj.weight.dtype)
+        #     new_proj.weight = nn.Parameter(torch.zeros_like(new_proj.weight))
+        #     new_proj.weight[:, :1024].copy_(model.wrapped.input_proj.weight)  #
+        #     new_proj.bias = nn.Parameter(torch.zeros_like(new_proj.bias))
+        #     new_proj.bias[:1024].copy_(model.wrapped.input_proj.bias)
+        #     model.wrapped.input_proj = new_proj
+        #
+        # ckp = torch.load(
+        #     hf_hub_download(repo_id='silentchen/Shap_Editor', subfolder='single', filename='{}.pt'.format(e_type)),
+        #     map_location='cpu')
+        # model.load_state_dict(ckp['model'])
 
         noise_initial = initial_noise[e_type].to(device)
         noise_start_t_e_type = noise_start_t[e_type]
@@ -324,8 +325,8 @@ def main():
         return mesh_path[0], mesh_path[1], mesh_path[2], mesh_path[3], state
 
 
-    del models
-    models = None
+    # del models
+    # models = None
     with Blocks(
             css=css,
             analytics_enabled=False,