diff options
| author | flamelaw <flamelaw.com3d2@gmail.com> | 2022-11-20 16:14:27 +0900 |
|---|---|---|
| committer | flamelaw <flamelaw.com3d2@gmail.com> | 2022-11-20 16:14:27 +0900 |
| commit | 2d22d72cdaaf2b78b2986b841d478c11ac855dd2 (patch) | |
| tree | 701812fcae555f62da6db7c672023ef5c9c7f455 | |
| parent | a4a5735d0a80218e59f8a6e8401726f7209a6a8d (diff) | |
fix random sampling with pin_memory
| -rw-r--r-- | modules/textual_inversion/dataset.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/textual_inversion/dataset.py b/modules/textual_inversion/dataset.py index 1dd53b85..110c0e09 100644 --- a/modules/textual_inversion/dataset.py +++ b/modules/textual_inversion/dataset.py @@ -134,7 +134,7 @@ class PersonalizedBase(Dataset): if self.tag_drop_out != 0 or self.shuffle_tags:
entry.cond_text = self.create_text(entry.filename_text)
if self.latent_sampling_method == "random":
- entry.latent_sample = shared.sd_model.get_first_stage_encoding(entry.latent_dist)
+ entry.latent_sample = shared.sd_model.get_first_stage_encoding(entry.latent_dist).to(devices.cpu)
return entry
class PersonalizedDataLoader(DataLoader):
|