Spaces:
Runtime error
Runtime error
Mehdi Cherti
commited on
Commit
·
ba46911
1
Parent(s):
301a4a3
update
Browse files
app.py
CHANGED
|
@@ -34,7 +34,6 @@ default = "diffusion_db_128ch_1timesteps_openclip_vith14"
|
|
| 34 |
def gen(md, model_name, md2, text, seed, nb_samples, width, height):
|
| 35 |
model = load(model_name)
|
| 36 |
torch.manual_seed(int(seed))
|
| 37 |
-
model = models[model_name]
|
| 38 |
nb_samples = int(nb_samples)
|
| 39 |
height = int(height)
|
| 40 |
width = int(width)
|
|
|
|
| 34 |
def gen(md, model_name, md2, text, seed, nb_samples, width, height):
|
| 35 |
model = load(model_name)
|
| 36 |
torch.manual_seed(int(seed))
|
|
|
|
| 37 |
nb_samples = int(nb_samples)
|
| 38 |
height = int(height)
|
| 39 |
width = int(width)
|