Update region_control.py

region_control.py

@@ -53,7 +53,8 @@ class MultiDiffusion(nn.Module):
         elif self.sd_version == '1.5':
             model_key = "runwayml/stable-diffusion-v1-5"
         else:
-            raise ValueError(f'Stable-diffusion version {self.sd_version} not supported.')
+            model_key = self.sd_version #For custom models or fine-tunes, allow people to use arbitrary versions
+            #raise ValueError(f'Stable-diffusion version {self.sd_version} not supported.')
 
         # Create model
         self.vae = AutoencoderKL.from_pretrained(model_key, subfolder="vae").to(self.device)