Views
No views yet
1import torch
2from diffusers import DiffusionPipeline
3from IPython.display import Audio
4
5
6device = "cuda" if torch.cuda.is_available() else "cpu"
7print(f"device: {device}")
8
9pipe = DiffusionPipeline.from_pretrained(
10 'Kevin3111/Electronic_test'
11).to(device)
12output = pipe(steps=50)
13
14display(output.images[0])
15display(Audio(output.audios[0], rate=pipe.mel.get_sample_rate()))