r/stable-diffusion-finetune
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

goooo

Browse source
This commit is contained in:
Patrick Esser 2022-06-01 09:14:09 +00:00 committed by root
parent 3769989f20
commit cb0414d89b
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml
View file

@ -110,11 +110,12 @@ lightning:
image_logger:
target: main.ImageLogger
params:
batch_frequency: 50
batch_frequency: 5000
max_images: 4
increase_log_steps: False
log_first_step: False
log_images_kwargs:
use_ema_scope: False
inpaint: False
plot_progressive_rows: False
plot_diffusion_rows: False

4
ldm/models/diffusion/ddpm.py
View file

@ -1194,7 +1194,9 @@ class LatentDiffusion(DDPM):
if isinstance(xc, dict) or isinstance(xc, list):
c = self.get_learned_conditioning(xc)
else:
c = self.get_learned_conditioning(xc.to(self.device))
if hasattr(xc, "to"):
xc = xc.to(self.device)
c = self.get_learned_conditioning(xc)
else:
# todo: get null label from cond_stage_model
raise NotImplementedError()