fix bug

hydra_node/models.py

@@ -350,6 +350,9 @@ class StableDiffusionModel(nn.Module):
             torch.seed()
             np.random.seed()
 
+        #set hypernetwork to none after generation
+        CrossAttention.set_hypernetwork(None)
+
         return images
 
     @torch.no_grad()

hydra_node/sanitize.py

@@ -104,7 +104,7 @@ def sanitize_stable_diffusion(request, config):
         random.setstate(state)
 
     if request.module is not None:
-        if request.module not in config.model.premodules:
+        if request.module not in config.model.premodules or request.module != "vanilla":
             return False, "module should be one of: " + ", ".join(config.model.premodules)
 
     if request.image is not None: