Hugging Face's logo

harmonai
/
jmann-large-580k

Diffusers
Safetensors
DanceDiffusionPipeline
audio-generation
Model card Files Community
1
patrickvonplaten commited on
Commit
59b7c37
·
1 Parent(s): 3bd9eda

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -7,6 +7,7 @@ import scipy
7
 
8
  model_id = "harmonai/jmann-large-580k"
9
  pipeline = DiffusionPipeline.from_pretrained(model_id)
 
10
 
11
  audio = pipeline(audio_length_in_s=4.0).audios
12
 
@@ -25,6 +26,7 @@ import torch
25
 
26
  model_id = "harmonai/jmann-large-580k"
27
  pipeline = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 
28
 
29
  audio = pipeline(audio_length_in_s=4.0).audios
30
 
 
7
 
8
  model_id = "harmonai/jmann-large-580k"
9
  pipeline = DiffusionPipeline.from_pretrained(model_id)
10
+ pipeline = pipeline.to("cuda")
11
 
12
  audio = pipeline(audio_length_in_s=4.0).audios
13
 
 
26
 
27
  model_id = "harmonai/jmann-large-580k"
28
  pipeline = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
29
+ pipeline = pipeline.to("cuda")
30
 
31
  audio = pipeline(audio_length_in_s=4.0).audios
32