Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -93,7 +93,7 @@ def sample_then_run(network, unet):
|
|
93 |
seed = 5
|
94 |
cfg = 3.0
|
95 |
steps = 25
|
96 |
-
image = inference( prompt, negative_prompt, cfg, steps, seed)
|
97 |
torch.save(network.proj, "model.pt" )
|
98 |
#return
|
99 |
return image, "model.pt", unet, network
|
|
|
93 |
seed = 5
|
94 |
cfg = 3.0
|
95 |
steps = 25
|
96 |
+
image = inference( network, unet, prompt, negative_prompt, cfg, steps, seed)
|
97 |
torch.save(network.proj, "model.pt" )
|
98 |
#return
|
99 |
return image, "model.pt", unet, network
|