Skip to content

Commit

Permalink
FIX
Browse files Browse the repository at this point in the history
  • Loading branch information
Dartvauder committed Nov 7, 2024
1 parent f4be475 commit 35990ed
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 9 deletions.
4 changes: 4 additions & 0 deletions .gitattributes
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
*.cpp linguist-vendored
*.h linguist-vendored
*.cu linguist-vendored
*.cuh linguist-vendored
11 changes: 2 additions & 9 deletions LaunchFile/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -11257,10 +11257,6 @@ def get_stable_diffusion_models():
]


audiocraft_models_list = [None] + ["musicgen-stereo-medium", "audiogen-medium", "musicgen-stereo-melody", "musicgen-medium", "musicgen-melody", "musicgen-large",
"hybrid-magnet-medium", "magnet-medium-30sec", "magnet-medium-10sec", "audio-magnet-medium"]


def get_vae_models():
return [None] + [
model for model in os.listdir("inputs/image/sd_models/vae")
Expand Down Expand Up @@ -11325,9 +11321,6 @@ def get_inpaint_models():
]


controlnet_models_list = [None, "openpose", "depth", "canny", "lineart", "scribble"]


def get_rvc_models():
return [
model_folder for model_folder in os.listdir("inputs/audio/rvc_models")
Expand Down Expand Up @@ -11823,7 +11816,7 @@ def create_footer():
"TCDScheduler", "DDIMScheduler", "PNDMScheduler", "DDPMScheduler"
], label=_("Select scheduler", lang), value="EulerDiscreteScheduler"),
gr.Dropdown(choices=model_lists["stable_diffusion"], label=_("Select StableDiffusion model", lang), value=None),
gr.Dropdown(choices=controlnet_models_list, label=_("Select ControlNet model", lang), value=None),
gr.Dropdown(choices=[None, "openpose", "depth", "canny", "lineart", "scribble"], label=_("Select ControlNet model", lang), value=None),
gr.Textbox(label=_("Seed (optional)", lang), value=""),
gr.Button(value=_("Stop generation", lang), interactive=True, variant="stop")
],
Expand Down Expand Up @@ -13584,7 +13577,7 @@ def create_footer():
inputs=[
gr.Textbox(label=_("Enter your prompt", lang)),
gr.Audio(type="filepath", label=_("Melody audio (optional)", lang), interactive=True),
gr.Dropdown(choices=audiocraft_models_list, label=_("Select AudioCraft model", lang), value=None),
gr.Dropdown(choices=[None, "musicgen-stereo-medium", "audiogen-medium", "musicgen-stereo-melody", "musicgen-medium", "musicgen-melody", "musicgen-large", "hybrid-magnet-medium", "magnet-medium-30sec", "magnet-medium-10sec", "audio-magnet-medium"], label=_("Select AudioCraft model", lang), value=None),
gr.Radio(choices=["musicgen", "audiogen", "magnet"], label=_("Select model type", lang), value="musicgen"),
],
additional_inputs=[
Expand Down

0 comments on commit 35990ed

Please sign in to comment.