Skip to content

Commit 1b331f4

Browse files
vdebortoHackable Diffusion Authors
vdeborto
authored and
Hackable Diffusion Authors
committed
Unify MARK comment casing to Title Case
PiperOrigin-RevId: 911369554
1 parent 36ea832 commit 1b331f4

16 files changed

Lines changed: 40 additions & 36 deletions

File tree

‎hackable_diffusion/kdiff/core.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@
5656
from kauldron import kd
5757

5858
################################################################################
59-
# MARK: Type aliases
59+
# MARK: Type Aliases
6060
################################################################################
6161

6262
Array = hd_typing.Array

‎hackable_diffusion/kdiff/evals.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626

2727

2828
################################################################################
29-
# MARK: Type aliases
29+
# MARK: Type Aliases
3030
################################################################################
3131

3232
Array = hd_typing.Array

‎hackable_diffusion/lib/architecture/arch_utils.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737
SkipConnectionMethod = arch_typing.SkipConnectionMethod
3838

3939
################################################################################
40-
# MARK: Reusable NN components
40+
# MARK: Reusable NN Components
4141
################################################################################
4242

4343
kernel_init = nn.initializers.lecun_normal()

‎hackable_diffusion/lib/architecture/attention.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727

2828

2929
################################################################################
30-
# MARK: Type aliases
30+
# MARK: Type Aliases
3131
################################################################################
3232

3333
Float = hd_typing.Float
@@ -45,7 +45,7 @@
4545
MASK_LOGITS_VALUE = -1e9
4646

4747
################################################################################
48-
# MARK: Attention utilities
48+
# MARK: Attention Utilities
4949
################################################################################
5050

5151

@@ -166,7 +166,7 @@ def _dot_product_attention(
166166

167167

168168
################################################################################
169-
# MARK: Multi-head attention
169+
# MARK: Multi-Head Attention
170170
################################################################################
171171

172172

‎hackable_diffusion/lib/architecture/conditioning_encoder.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
EmbeddingMergeMethod = arch_typing.EmbeddingMergeMethod
4545

4646
################################################################################
47-
# MARK: Base classes
47+
# MARK: Base Classes
4848
################################################################################
4949

5050

@@ -81,7 +81,7 @@ def __call__(
8181

8282

8383
################################################################################
84-
# MARK: Time embedders
84+
# MARK: Time Embedders
8585
################################################################################
8686

8787

@@ -162,7 +162,7 @@ def __call__(self, time: hd_typing.TimeArray) -> hd_typing.TimeArray:
162162

163163

164164
################################################################################
165-
# MARK: Conditioning embedders
165+
# MARK: Conditioning Embedders
166166
################################################################################
167167

168168

@@ -338,7 +338,7 @@ def __call__(
338338

339339

340340
################################################################################
341-
# MARK: Process and combine time and conditioning signals
341+
# MARK: Process and Combine Time and Conditioning Signals
342342
################################################################################
343343

344344

‎hackable_diffusion/lib/architecture/sequence_embedders.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626

2727

2828
################################################################################
29-
# MARK: Type aliases
29+
# MARK: Type Aliases
3030
################################################################################
3131

3232
Float = hd_typing.Float
@@ -36,7 +36,7 @@
3636
RoPEPositionType = arch_typing.RoPEPositionType
3737

3838
################################################################################
39-
# MARK: Sequence embedding modules
39+
# MARK: Sequence Embedding Modules
4040
################################################################################
4141

4242

‎hackable_diffusion/lib/architecture/unet.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
import kauldron.ktyping as kt
2828

2929
################################################################################
30-
# MARK: Common types
30+
# MARK: Type Aliases
3131
################################################################################
3232

3333
DType = hd_typing.DType

‎hackable_diffusion/lib/architecture/unet_blocks.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
import kauldron.ktyping as kt
2626

2727
################################################################################
28-
# MARK: Common types and aliases
28+
# MARK: Type Aliases
2929
################################################################################
3030

3131
DType = hd_typing.DType
@@ -49,7 +49,7 @@
4949
DownsampleOutput = Float["batch height/2 width/2 output_channels"]
5050

5151
################################################################################
52-
# MARK: Input and Output blocks
52+
# MARK: Input and Output Blocks
5353
################################################################################
5454

5555

@@ -121,7 +121,7 @@ def __call__(self, x: BaseInput) -> BaseOutput:
121121

122122

123123
################################################################################
124-
# MARK: Residual block with optional resampling
124+
# MARK: Residual Block With Optional Resampling
125125
################################################################################
126126

127127

@@ -218,7 +218,7 @@ def __call__(
218218

219219

220220
################################################################################
221-
# MARK: Attention residual block
221+
# MARK: Attention Residual Block
222222
################################################################################
223223

224224

‎hackable_diffusion/lib/corruption/discrete.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ class SamplingPrecisionMode(enum.StrEnum):
6666

6767

6868
################################################################################
69-
# MARK: Projection functions
69+
# MARK: Projection Functions
7070
################################################################################
7171

7272

@@ -333,7 +333,7 @@ def get_schedule_info(self, time: TimeArray) -> dict[str, TimeArray]:
333333
return self.schedule.evaluate(time)
334334

335335
##############################################################################
336-
# MARK: Factory methods
336+
# MARK: Factory Methods
337337
##############################################################################
338338

339339
@classmethod

‎hackable_diffusion/lib/corruption/gaussian.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -270,7 +270,7 @@ def _get_alpha_sigma_and_der(
270270
################################################################################
271271

272272
################################################################################
273-
# MARK: convert from x0
273+
# MARK: Convert From x0
274274
################################################################################
275275

276276

@@ -296,7 +296,7 @@ def x0_to_v(x0, xt, alpha, sigma, alpha_der, sigma_der):
296296

297297

298298
################################################################################
299-
# MARK: convert from epsilon
299+
# MARK: Convert From Epsilon
300300
################################################################################
301301

302302

@@ -322,7 +322,7 @@ def epsilon_to_v(epsilon, xt, alpha, sigma, alpha_der, sigma_der):
322322

323323

324324
################################################################################
325-
# MARK: convert from score
325+
# MARK: Convert From Score
326326
################################################################################
327327

328328

@@ -354,7 +354,7 @@ def score_to_v(score, xt, alpha, sigma, alpha_der, sigma_der):
354354

355355

356356
################################################################################
357-
# MARK: convert from velocity
357+
# MARK: Convert From Velocity
358358
################################################################################
359359

360360

@@ -396,7 +396,7 @@ def velocity_to_v(velocity, xt, alpha, sigma, alpha_der, sigma_der):
396396

397397

398398
################################################################################
399-
# MARK: convert from v
399+
# MARK: Convert From v
400400
################################################################################
401401

402402

@@ -443,7 +443,7 @@ def v_to_velocity(v, xt, alpha, sigma, alpha_der, sigma_der):
443443

444444

445445
################################################################################
446-
# MARK: helpers
446+
# MARK: Helpers
447447
################################################################################
448448

449449

0 commit comments

Comments
 (0)