Skip to content

GitLab

Unverified Commit 1093f9d6 authored by Steven Liu's avatar Steven Liu Committed by GitHub
Browse files

[docs] MusicLDM (#5854) 

* fix

* feedback
parent 81780882
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/musicldm/pipeline_musicldm.py
View file @ 1093f9d6
...@@ -51,7 +51,7 @@ EXAMPLE_DOC_STRING = """ ...@@ -51,7 +51,7 @@ EXAMPLE_DOC_STRING = """
>>> import torch >>> import torch
>>> import scipy >>> import scipy
>>> repo_id = "cvssp/audioldm-s-full-v2" >>> repo_id = "ucsd-reach/musicldm"
>>> pipe = MusicLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float16) >>> pipe = MusicLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float16)
>>> pipe = pipe.to("cuda") >>> pipe = pipe.to("cuda")
......
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