patrickvonplaten commited on
Commit
fd9b508
1 Parent(s): d0b04f4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -14,10 +14,10 @@ from diffusers import DiffusionPipeline
14
  from scipy.io.wavfile import write
15
 
16
  model_id = "harmonai/maestro-150k"
17
- pipeline = DiffusionPipeline.from_pretrained(model_id)
18
- pipeline = pipeline.to("cuda")
19
 
20
- audios = pipeline(audio_length_in_s=4.0).audios
21
 
22
  # To save locally
23
  for i, audio in enumerate(audios):
@@ -40,8 +40,8 @@ from scipy.io.wavfile import write
40
  import torch
41
 
42
  model_id = "harmonai/maestro-150k"
43
- pipeline = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
44
- pipeline = pipeline.to("cuda")
45
 
46
  audios = pipeline(audio_length_in_s=4.0).audios
47
 
 
14
  from scipy.io.wavfile import write
15
 
16
  model_id = "harmonai/maestro-150k"
17
+ pipe = DiffusionPipeline.from_pretrained(model_id)
18
+ pipe = pipe.to("cuda")
19
 
20
+ audios = pipe(audio_length_in_s=4.0).audios
21
 
22
  # To save locally
23
  for i, audio in enumerate(audios):
 
40
  import torch
41
 
42
  model_id = "harmonai/maestro-150k"
43
+ pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
44
+ pipe = pipe.to("cuda")
45
 
46
  audios = pipeline(audio_length_in_s=4.0).audios
47