uripper commited on
Commit
636451d
·
1 Parent(s): a4ed3ea

changes to scheduler

Browse files
Files changed (2) hide show
  1. GIANNIS +1 -1
  2. app.py +1 -1
GIANNIS CHANGED
@@ -1 +1 @@
1
- Subproject commit 7a0f21e34b5fc9d30ada99349d6be6fed9301845
 
1
+ Subproject commit 7cc7631b6ff232e32f34e673ed05f9a01f20d26b
app.py CHANGED
@@ -3,7 +3,7 @@ from diffusers import PNDMPipeline, PNDMScheduler
3
  import random
4
 
5
 
6
- scheduler = PNDMScheduler(num_train_timesteps=1000,beta_schedule="linear")
7
  pipeline = PNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler)
8
  # num_imgs = len(os.listdir("\stored_images"))
9
  # imgs = os.listdir("\stored_images")
 
3
  import random
4
 
5
 
6
+ scheduler = PNDMScheduler(num_train_timesteps=50,beta_schedule="linear")
7
  pipeline = PNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler)
8
  # num_imgs = len(os.listdir("\stored_images"))
9
  # imgs = os.listdir("\stored_images")