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
|