Spaces:
Runtime error
Runtime error
Commit
•
b1ec36c
1
Parent(s):
e496e33
minor
Browse files
app.py
CHANGED
|
@@ -51,6 +51,7 @@ def load_model(args):
|
|
| 51 |
|
| 52 |
|
| 53 |
def infer(cfg_scale, top_k, top_p, temperature, class_label, seed):
|
|
|
|
| 54 |
n = 4
|
| 55 |
latent_size = image_size // args.downsample_size
|
| 56 |
# Labels to condition the model with (feel free to change):
|
|
|
|
| 51 |
|
| 52 |
|
| 53 |
def infer(cfg_scale, top_k, top_p, temperature, class_label, seed):
|
| 54 |
+
args.cfg_scale = cfg_scale
|
| 55 |
n = 4
|
| 56 |
latent_size = image_size // args.downsample_size
|
| 57 |
# Labels to condition the model with (feel free to change):
|