Skip to content

Commit c2cb173

Browse files
committed
Removed commit hash logic
1 parent 26be21b commit c2cb173

1 file changed

Lines changed: 2 additions & 9 deletions

File tree

src/maxdiffusion/generate_wan.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -161,18 +161,12 @@ def inference_generate_video(config, pipeline, filename_prefix=""):
161161
return
162162

163163

164-
def run(config, pipeline=None, filename_prefix="", commit_hash=None):
164+
def run(config, pipeline=None, filename_prefix=""):
165165
model_key = config.model_name
166166
writer = max_utils.initialize_summary_writer(config)
167167
if jax.process_index() == 0 and writer:
168168
max_logging.log(f"TensorBoard logs will be written to: {config.tensorboard_dir}")
169169

170-
if commit_hash:
171-
writer.add_text("inference/git_commit_hash", commit_hash, global_step=0)
172-
max_logging.log(f"Git Commit Hash: {commit_hash}")
173-
else:
174-
max_logging.log("Could not retrieve Git commit hash.")
175-
176170
if pipeline is None:
177171
model_type = config.model_type
178172
if model_key == WAN2_1:
@@ -248,13 +242,12 @@ def run(config, pipeline=None, filename_prefix="", commit_hash=None):
248242

249243

250244
def main(argv: Sequence[str]) -> None:
251-
commit_hash = get_git_commit_hash()
252245
pyconfig.initialize(argv)
253246
try:
254247
flax.config.update("flax_always_shard_variable", False)
255248
except LookupError:
256249
pass
257-
run(pyconfig.config, commit_hash=commit_hash)
250+
run(pyconfig.config)
258251

259252

260253
if __name__ == "__main__":

0 commit comments

Comments
 (0)