diff --git a/StabilityMatrix/Models/CheckpointFile.cs b/StabilityMatrix/Models/CheckpointFile.cs index ad83f1ad..0a80b117 100644 --- a/StabilityMatrix/Models/CheckpointFile.cs +++ b/StabilityMatrix/Models/CheckpointFile.cs @@ -39,7 +39,7 @@ public partial class CheckpointFile : ObservableObject public string FileName => Path.GetFileName(FilePath); - private static readonly string[] SupportedCheckpointExtensions = { ".safetensors", ".pt", ".ckpt", ".pth" }; + private static readonly string[] SupportedCheckpointExtensions = { ".safetensors", ".pt", ".ckpt", ".pth", "bin" }; private static readonly string[] SupportedImageExtensions = { ".png", ".jpg", ".jpeg" }; partial void OnConnectedModelChanged(ConnectedModelInfo? value) diff --git a/StabilityMatrix/Models/Packages/ComfyUI.cs b/StabilityMatrix/Models/Packages/ComfyUI.cs index bbce4fab..ca28830f 100644 --- a/StabilityMatrix/Models/Packages/ComfyUI.cs +++ b/StabilityMatrix/Models/Packages/ComfyUI.cs @@ -30,6 +30,22 @@ public class ComfyUI : BaseGitPackage { } + // https://github.com/comfyanonymous/ComfyUI/blob/master/folder_paths.py#L11 + public override Dictionary SharedFolders => new() + { + [SharedFolderType.StableDiffusion] = "models/checkpoints", + [SharedFolderType.Diffusers] = "models/diffusers", + [SharedFolderType.Lora] = "models/loras", + [SharedFolderType.CLIP] = "models/clip", + [SharedFolderType.TextualInversion] = "models/embeddings", + [SharedFolderType.VAE] = "models/vae", + [SharedFolderType.ApproxVAE] = "models/vae_approx", + [SharedFolderType.ControlNet] = "models/controlnet", + [SharedFolderType.GLIGEN] = "models/gligen", + [SharedFolderType.ESRGAN] = "models/upscale_models", + [SharedFolderType.Hypernetwork] = "models/hypernetworks", + }; + public override List LaunchOptions => new() { new() diff --git a/StabilityMatrix/Models/Packages/VladAutomatic.cs b/StabilityMatrix/Models/Packages/VladAutomatic.cs index 61836888..b29d3ed2 100644 --- a/StabilityMatrix/Models/Packages/VladAutomatic.cs +++ b/StabilityMatrix/Models/Packages/VladAutomatic.cs @@ -25,6 +25,27 @@ public class VladAutomatic : BaseGitPackage base(githubApi, settingsManager, downloadService) { } + + // https://github.com/vladmandic/automatic/blob/master/modules/shared.py#L324 + public override Dictionary SharedFolders => new() + { + [SharedFolderType.StableDiffusion] = "models/Stable-diffusion", + [SharedFolderType.Diffusers] = "models/Diffusers", + [SharedFolderType.VAE] = "models/VAE", + [SharedFolderType.TextualInversion] = "models/embeddings", + [SharedFolderType.Hypernetwork] = "models/hypernetworks", + [SharedFolderType.Codeformer] = "models/Codeformer", + [SharedFolderType.GFPGAN] = "models/GFPGAN", + [SharedFolderType.BSRGAN] = "models/BSRGAN", + [SharedFolderType.ESRGAN] = "models/ESRGAN", + [SharedFolderType.RealESRGAN] = "models/RealESRGAN", + [SharedFolderType.ScuNET] = "models/ScuNET", + [SharedFolderType.SwinIR] = "models/SwinIR", + [SharedFolderType.LDSR] = "models/LDSR", + [SharedFolderType.CLIP] = "models/CLIP", + [SharedFolderType.Lora] = "models/Lora", + [SharedFolderType.LyCORIS] = "models/LyCORIS", + }; public override List LaunchOptions => new() { diff --git a/StabilityMatrix/Models/SharedFolderType.cs b/StabilityMatrix/Models/SharedFolderType.cs index c9c3c51c..08aa8544 100644 --- a/StabilityMatrix/Models/SharedFolderType.cs +++ b/StabilityMatrix/Models/SharedFolderType.cs @@ -10,6 +10,10 @@ public enum SharedFolderType Lora, LyCORIS, ESRGAN, + GFPGAN, + BSRGAN, + Codeformer, + Diffusers, RealESRGAN, SwinIR, VAE, @@ -18,5 +22,9 @@ public enum SharedFolderType DeepDanbooru, TextualInversion, Hypernetwork, - ControlNet + ControlNet, + LDSR, + CLIP, + ScuNET, + GLIGEN, }