Unverified Commit ea1b4ea7 authored by Yoach Lacombe's avatar Yoach Lacombe Committed by GitHub
Browse files

Fix Stable Audio repository id (#9016)

Fix Stable Audio repo id
parent e5b94b4c
......@@ -44,7 +44,7 @@ EXAMPLE_DOC_STRING = """
>>> import soundfile as sf
>>> from diffusers import StableAudioPipeline
>>> repo_id = "ylacombe/stable-audio-1.0" # TODO (YL): change once set
>>> repo_id = "stabilityai/stable-audio-open-1.0"
>>> pipe = StableAudioPipeline.from_pretrained(repo_id, torch_dtype=torch.float16)
>>> pipe = pipe.to("cuda")
......
......@@ -1182,9 +1182,7 @@ class AutoencoderOobleckIntegrationTests(unittest.TestCase):
return audio
def get_oobleck_vae_model(
self, model_id="ylacombe/stable-audio-1.0", fp16=False
): # TODO (YL): change repo id once moved
def get_oobleck_vae_model(self, model_id="stabilityai/stable-audio-open-1.0", fp16=False):
torch_dtype = torch.float16 if fp16 else torch.float32
model = AutoencoderOobleck.from_pretrained(
......
......@@ -439,9 +439,7 @@ class StableAudioPipelineIntegrationTests(unittest.TestCase):
return inputs
def test_stable_audio(self):
stable_audio_pipe = StableAudioPipeline.from_pretrained(
"ylacombe/stable-audio-1.0"
) # TODO (YL): change once changed
stable_audio_pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0")
stable_audio_pipe = stable_audio_pipe.to(torch_device)
stable_audio_pipe.set_progress_bar_config(disable=None)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment