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)
|