Spaces:

piyushgrover
/

Stable-Diffusion-Image-Generation

Runtime error

piyushgrover commited on Oct 20, 2023

Commit

7a5d971

1 Parent(s): bc7673a

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -46,6 +46,7 @@ vae = vae.to(torch_device)
 text_encoder = text_encoder.to(torch_device)
 unet = unet.to(torch_device)
 def load_learned_embeds():
     pathlist = Path('learned_embeds/').glob('*_learned_embeds.bin')

 text_encoder = text_encoder.to(torch_device)
 unet = unet.to(torch_device)
+seed_values = [0, 0, 0, 0, 0]
 def load_learned_embeds():
     pathlist = Path('learned_embeds/').glob('*_learned_embeds.bin')