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
|