diff options
| -rw-r--r-- | modules/sd_samplers_common.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py index 42a29fc9..39586b40 100644 --- a/modules/sd_samplers_common.py +++ b/modules/sd_samplers_common.py @@ -39,7 +39,7 @@ def samples_to_images_tensor(sample, approximation=None, model=None): else:
if model is None:
model = shared.sd_model
- x_sample = model.decode_first_stage(sample)
+ x_sample = model.decode_first_stage(sample.to(model.first_stage_model.dtype))
return x_sample
|