theadamsabra commited on
Commit
e377745
1 Parent(s): 89d45c5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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/al_nay_diffusion"
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