Change utils dependency

video2world_hf.py

@@ -1,33 +1,17 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
-# SPDX-License-Identifier: Apache-2.0
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
 import argparse
 import os
 
-from .utils import misc
+# from .utils import misc
 import torch
 
 from .inference_utils import add_common_arguments, check_input_frames, validate_args
 from .world_generation_pipeline import DiffusionVideo2WorldGenerationPipeline
-from .utils import log
-from .utils.utils_io import read_prompts_from_file, save_video
+from .log import log
+from .utils_io import read_prompts_from_file, save_video
 
 from .download_diffusion import main as download_diffusion
 from transformers import PreTrainedModel, PretrainedConfig
 
-torch.enable_grad(False)
 
 #custom config class
 class DiffusionVideo2WorldConfig(PretrainedConfig):
@@ -65,9 +49,10 @@ class DiffusionVideo2World(PreTrainedModel):
 
     def __init__(self, config=DiffusionVideo2WorldConfig()):
         super().__init__(config)
+        torch.enable_grad(False)
         cfg = config
-
-        misc.set_random_seed(cfg.seed)
+        self.config = config
+        # misc.set_random_seed(cfg.seed)
         inference_type = "video2world"
         validate_args(cfg, inference_type)