Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
ea1b4ea7
Unverified
Commit
ea1b4ea7
authored
Jul 30, 2024
by
Yoach Lacombe
Committed by
GitHub
Jul 30, 2024
Browse files
Fix Stable Audio repository id (#9016)
Fix Stable Audio repo id
parent
e5b94b4c
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
7 deletions
+3
-7
src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py
...diffusers/pipelines/stable_audio/pipeline_stable_audio.py
+1
-1
tests/models/autoencoders/test_models_vae.py
tests/models/autoencoders/test_models_vae.py
+1
-3
tests/pipelines/stable_audio/test_stable_audio.py
tests/pipelines/stable_audio/test_stable_audio.py
+1
-3
No files found.
src/diffusers/pipelines/stable_audio/pipeline_stable_audio.py
View file @
ea1b4ea7
...
...
@@ -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")
...
...
tests/models/autoencoders/test_models_vae.py
View file @
ea1b4ea7
...
...
@@ -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
(
...
...
tests/pipelines/stable_audio/test_stable_audio.py
View file @
ea1b4ea7
...
...
@@ -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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment