Spaces:
Runtime error
Runtime error
Commit
·
0bea37e
1
Parent(s):
2c28657
Update utils.py
Browse files
utils.py
CHANGED
@@ -39,14 +39,14 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
|
|
39 |
c = model.get_learned_conditioning(c_dict)
|
40 |
c = model.enc_concat_seq(c, c_dict, 'c_concat')
|
41 |
|
42 |
-
samples_ddim, _ = sampler.sample(S=
|
43 |
conditioning=c,
|
44 |
batch_size=1,
|
45 |
shape=[3, 64, 64],
|
46 |
-
verbose=False,
|
47 |
-
unconditional_guidance_scale=5.0,
|
48 |
-
unconditional_conditioning=uc,
|
49 |
-
eta=0)
|
50 |
|
51 |
x_samples_ddim = model.decode_first_stage(samples_ddim)
|
52 |
#x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0)
|
|
|
39 |
c = model.get_learned_conditioning(c_dict)
|
40 |
c = model.enc_concat_seq(c, c_dict, 'c_concat')
|
41 |
|
42 |
+
samples_ddim, _ = sampler.sample(S=999,
|
43 |
conditioning=c,
|
44 |
batch_size=1,
|
45 |
shape=[3, 64, 64],
|
46 |
+
verbose=False, eta=1)
|
47 |
+
#unconditional_guidance_scale=5.0,
|
48 |
+
#unconditional_conditioning=uc,
|
49 |
+
#eta=0)
|
50 |
|
51 |
x_samples_ddim = model.decode_first_stage(samples_ddim)
|
52 |
#x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0)
|