diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-31 07:37:36 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-31 07:37:36 +0300 |
| commit | 20158d77d9fa37dbbd38e5a281e29560caf4a337 (patch) | |
| tree | 7761c9fecfd8ca6656d56c5300181a5ec3d1a98a /modules/sd_samplers_common.py | |
| parent | 3bff988f1e66a215b4795be012c868d06e0417c9 (diff) | |
| parent | e7965a5eb804a51e949df07c66c0b7c61ab7fa7b (diff) | |
Merge branch 'release_candidate' into dev
Diffstat (limited to 'modules/sd_samplers_common.py')
| -rw-r--r-- | modules/sd_samplers_common.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py index 6c935a38..58efcad2 100644 --- a/modules/sd_samplers_common.py +++ b/modules/sd_samplers_common.py @@ -95,6 +95,8 @@ def images_tensor_to_samples(image, approximation=None, model=None): else:
if model is None:
model = shared.sd_model
+ model.first_stage_model.to(devices.dtype_vae)
+
image = image.to(shared.device, dtype=devices.dtype_vae)
image = image * 2 - 1
if len(image) > 1:
|