uripper
commited on
Commit
·
aa8e46c
1
Parent(s):
636451d
improving timesteps
Browse files
app.py
CHANGED
|
@@ -3,7 +3,7 @@ from diffusers import PNDMPipeline, PNDMScheduler
|
|
| 3 |
import random
|
| 4 |
|
| 5 |
|
| 6 |
-
scheduler = PNDMScheduler(num_train_timesteps=
|
| 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=500,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")
|