Skip to content

Commit 864999b

Browse files
committed
remove unused files
1 parent fc2bae7 commit 864999b

File tree

3 files changed

+0
-251
lines changed

3 files changed

+0
-251
lines changed

model/blocks.py

-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
from config_base import BaseConfig
1010
from torch import nn
1111

12-
from .fp16_util import convert_module_to_f16, convert_module_to_f32
1312
from .nn import (avg_pool_nd, conv_nd, linear, normalization,
1413
timestep_embedding, torch_checkpoint, zero_module)
1514

model/fp16_util.py

-236
This file was deleted.

model/unet.py

-14
Original file line numberDiff line numberDiff line change
@@ -493,20 +493,6 @@ def __init__(self, conf: BeatGANsEncoderConfig):
493493
else:
494494
raise NotImplementedError(f"Unexpected {conf.pool} pooling")
495495

496-
def convert_to_fp16(self):
497-
"""
498-
Convert the torso of the model to float16.
499-
"""
500-
self.input_blocks.apply(convert_module_to_f16)
501-
self.middle_block.apply(convert_module_to_f16)
502-
503-
def convert_to_fp32(self):
504-
"""
505-
Convert the torso of the model to float32.
506-
"""
507-
self.input_blocks.apply(convert_module_to_f32)
508-
self.middle_block.apply(convert_module_to_f32)
509-
510496
def forward(self, x, t=None, return_2d_feature=False):
511497
"""
512498
Apply the model to an input batch.

0 commit comments

Comments
 (0)