Spaces:

stevhliu
/

unconditional-image-generation

Running

stevhliu HF Staff commited on Nov 22, 2023

Commit

75212fe

1 Parent(s): cd02dc8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 import torch
 from diffusers import DiffusionPipeline
-pipeline = DiffusionPipeline.from_pretrained("anton-l/ddpm-butterflies-128")
 def diffusion():
     images = []

 import torch
 from diffusers import DiffusionPipeline
+pipeline = DiffusionPipeline.from_pretrained("anton-l/ddpm-butterflies-128", use_safetensors=True)
 def diffusion():
     images = []