theadamsabra
commited on
Commit
•
e377745
1
Parent(s):
89d45c5
Update README.md
Browse files
README.md
CHANGED
@@ -21,7 +21,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
21 |
generator = torch.Generator(device=device)
|
22 |
|
23 |
# Instantiate model
|
24 |
-
model_id = "mijwiz-laboratories/
|
25 |
audio_diffusion = DiffusionPipeline.from_pretrained(model_id).to(device)
|
26 |
|
27 |
# Set seed for generator
|
|
|
21 |
generator = torch.Generator(device=device)
|
22 |
|
23 |
# Instantiate model
|
24 |
+
model_id = "mijwiz-laboratories/al_nay_diffusion_unconditional_256"
|
25 |
audio_diffusion = DiffusionPipeline.from_pretrained(model_id).to(device)
|
26 |
|
27 |
# Set seed for generator
|