diff options
| -rw-r--r-- | modules/prompt_parser.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/prompt_parser.py b/modules/prompt_parser.py index 67a4683a..a6a25b28 100644 --- a/modules/prompt_parser.py +++ b/modules/prompt_parser.py @@ -114,7 +114,7 @@ def get_learned_conditioning(prompts, steps): def reconstruct_cond_batch(c: ScheduledPromptBatch, current_step):
- res = torch.zeros(c.shape, device=shared.device, dtype=torch.half)
+ res = torch.zeros(c.shape, device=shared.device, dtype=next(shared.sd_model.parameters()).dtype)
for i, cond_schedule in enumerate(c.schedules):
target_index = 0
for curret_index, (end_at, cond) in enumerate(cond_schedule):
|