Skip to content

Commit 41187c9

Browse files
committed
update
1 parent cea2380 commit 41187c9

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

scripts/4-4_stable-diffusion-v1.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -333,6 +333,7 @@ def sample(
333333
# %% [markdown]
334334
# ### プロンプト埋め込みを取得する関数の定義
335335

336+
336337
# %%
337338
@torch.no_grad()
338339
def encode_prompt_embeds(
@@ -396,6 +397,7 @@ def apply_cfg(
396397
# %% [markdown]
397398
# ### 逆拡散過程の実行関数の定義
398399

400+
399401
# %%
400402
@torch.no_grad()
401403
def reverse_diffusion_step(
@@ -450,6 +452,7 @@ def reverse_diffusion_step(
450452
# %% [markdown]
451453
# ### 潜在表現から画像への変換関数の定義
452454

455+
453456
# %%
454457
@torch.no_grad()
455458
def decode_latents_to_images(
@@ -517,6 +520,7 @@ def decode_latents_to_images(
517520
# %% [markdown]
518521
# ### Inversion 実行関数の定義
519522

523+
520524
# %%
521525
def invert(
522526
prompt: str,
@@ -571,6 +575,7 @@ def invert(
571575
# %% [markdown]
572576
# ### 1 ステップの Inversion を実行する関数の定義
573577

578+
574579
# %%
575580
@torch.no_grad()
576581
def invert_step(

0 commit comments

Comments
 (0)