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 = """ ...@@ -44,7 +44,7 @@ EXAMPLE_DOC_STRING = """
>>> import soundfile as sf >>> import soundfile as sf
>>> from diffusers import StableAudioPipeline >>> 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 = StableAudioPipeline.from_pretrained(repo_id, torch_dtype=torch.float16)
>>> pipe = pipe.to("cuda") >>> pipe = pipe.to("cuda")
......
...@@ -1182,9 +1182,7 @@ class AutoencoderOobleckIntegrationTests(unittest.TestCase): ...@@ -1182,9 +1182,7 @@ class AutoencoderOobleckIntegrationTests(unittest.TestCase):
return audio return audio
def get_oobleck_vae_model( def get_oobleck_vae_model(self, model_id="stabilityai/stable-audio-open-1.0", fp16=False):
self, model_id="ylacombe/stable-audio-1.0", fp16=False
): # TODO (YL): change repo id once moved
torch_dtype = torch.float16 if fp16 else torch.float32 torch_dtype = torch.float16 if fp16 else torch.float32
model = AutoencoderOobleck.from_pretrained( model = AutoencoderOobleck.from_pretrained(
......
...@@ -439,9 +439,7 @@ class StableAudioPipelineIntegrationTests(unittest.TestCase): ...@@ -439,9 +439,7 @@ class StableAudioPipelineIntegrationTests(unittest.TestCase):
return inputs return inputs
def test_stable_audio(self): def test_stable_audio(self):
stable_audio_pipe = StableAudioPipeline.from_pretrained( stable_audio_pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0")
"ylacombe/stable-audio-1.0"
) # TODO (YL): change once changed
stable_audio_pipe = stable_audio_pipe.to(torch_device) stable_audio_pipe = stable_audio_pipe.to(torch_device)
stable_audio_pipe.set_progress_bar_config(disable=None) 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