Spaces:
Running
on
Zero
Running
on
Zero
amildravid4292
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -41,6 +41,10 @@ def sample_model():
|
|
41 |
unet, _, _, _, _ = load_models(device)
|
42 |
network = sample_weights(unet, proj, mean, std, v[:, :1000], device, factor = 1.00)
|
43 |
|
|
|
|
|
|
|
|
|
44 |
|
45 |
@torch.no_grad()
|
46 |
def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
|
|
|
41 |
unet, _, _, _, _ = load_models(device)
|
42 |
network = sample_weights(unet, proj, mean, std, v[:, :1000], device, factor = 1.00)
|
43 |
|
44 |
+
### start off with an initial model
|
45 |
+
sample_weights()
|
46 |
+
|
47 |
+
|
48 |
|
49 |
@torch.no_grad()
|
50 |
def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
|