This commit is contained in:
root 2022-06-08 21:15:19 +00:00
parent 5d40de00fb
commit 676e6acfef

View file

@ -9,7 +9,7 @@ model:
timesteps: 1000
first_stage_key: "jpg"
cond_stage_key: "txt"
image_size: 32
image_size: 64
channels: 4
cond_stage_trainable: false # Note: different from the one we trained before
conditioning_key: crossattn
@ -28,7 +28,7 @@ model:
unet_config:
target: ldm.modules.diffusionmodules.openaimodel.UNetModel
params:
image_size: 32
image_size: 32 # unused
in_channels: 4
out_channels: 4
model_channels: 320