Spaces:

1aurent
/

ddpm-mnist

Sleeping

stevhliu HF staff commited on Mar 7, 2023

Commit

b84ee6b

1 Parent(s): 5f91254

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,6 @@ import random
 import numpy as np
 pipeline = DiffusionPipeline.from_pretrained("johnowhitaker/ddpm-butterflies-32px")
-pipeline.to('cuda')
 def predict(steps, seed):
     generator = torch.manual_seed(seed)

 import numpy as np
 pipeline = DiffusionPipeline.from_pretrained("johnowhitaker/ddpm-butterflies-32px")
 def predict(steps, seed):
     generator = torch.manual_seed(seed)