AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Fix Stable Audio repository id (#9016)

Browse Source 
Fix Stable Audio repo id
This commit is contained in:
Yoach Lacombe
2024-07-30 19:47:44 +02:00
committed by GitHub
parent e5b94b4c57
commit ea1b4ea7ca
3 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py
View File

@@ -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")

4
tests/models/autoencoders/test_models_vae.py
View File

@@ -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(

4
tests/pipelines/stable_audio/test_stable_audio.py
View File

@@ -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)