Skip to content

Commit ea1b4ea

Browse files
authored
Fix Stable Audio repository id (#9016)
Fix Stable Audio repo id
1 parent e5b94b4 commit ea1b4ea

File tree

3 files changed

+3
-7
lines changed

3 files changed

+3
-7
lines changed

src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
>>> import soundfile as sf
4545
>>> from diffusers import StableAudioPipeline
4646
47-
>>> repo_id = "ylacombe/stable-audio-1.0" # TODO (YL): change once set
47+
>>> repo_id = "stabilityai/stable-audio-open-1.0"
4848
>>> pipe = StableAudioPipeline.from_pretrained(repo_id, torch_dtype=torch.float16)
4949
>>> pipe = pipe.to("cuda")
5050

tests/models/autoencoders/test_models_vae.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1182,9 +1182,7 @@ def get_audio(self, audio_sample_size=2097152, fp16=False):
11821182

11831183
return audio
11841184

1185-
def get_oobleck_vae_model(
1186-
self, model_id="ylacombe/stable-audio-1.0", fp16=False
1187-
): # TODO (YL): change repo id once moved
1185+
def get_oobleck_vae_model(self, model_id="stabilityai/stable-audio-open-1.0", fp16=False):
11881186
torch_dtype = torch.float16 if fp16 else torch.float32
11891187

11901188
model = AutoencoderOobleck.from_pretrained(

tests/pipelines/stable_audio/test_stable_audio.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -439,9 +439,7 @@ def get_inputs(self, device, generator_device="cpu", dtype=torch.float32, seed=0
439439
return inputs
440440

441441
def test_stable_audio(self):
442-
stable_audio_pipe = StableAudioPipeline.from_pretrained(
443-
"ylacombe/stable-audio-1.0"
444-
) # TODO (YL): change once changed
442+
stable_audio_pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0")
445443
stable_audio_pipe = stable_audio_pipe.to(torch_device)
446444
stable_audio_pipe.set_progress_bar_config(disable=None)
447445

0 commit comments

Comments
 (0)