EthanZyh commited on
Commit
f9c7c59
·
1 Parent(s): 645644c

try multiple files import at root dir

Browse files
Files changed (3) hide show
  1. testimport1.py +2 -0
  2. testimport2.py +1 -0
  3. text2world_hf.py +7 -5
testimport1.py ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ from testimport2 import testB
2
+ testA = 1
testimport2.py ADDED
@@ -0,0 +1 @@
 
 
1
+ testB = 2
text2world_hf.py CHANGED
@@ -3,11 +3,13 @@ import argparse
3
  import torch
4
  from transformers import PreTrainedModel, PretrainedConfig
5
 
6
- from .cosmos1.models.diffusion.inference.inference_utils import add_common_arguments, validate_args
7
- from .cosmos1.models.diffusion.inference.world_generation_pipeline import DiffusionText2WorldGenerationPipeline
8
- import .cosmos1.utils.log as log
9
- import .cosmos1.utils.misc as misc
10
- from .cosmos1.utils.io import read_prompts_from_file, save_video
 
 
11
 
12
  class DiffusionText2WorldConfig(PretrainedConfig):
13
  model_type = "DiffusionText2World"
 
3
  import torch
4
  from transformers import PreTrainedModel, PretrainedConfig
5
 
6
+ from .testimport1 import testA, testB
7
+
8
+ # from .cosmos1.models.diffusion.inference.inference_utils import add_common_arguments, validate_args
9
+ # from .cosmos1.models.diffusion.inference.world_generation_pipeline import DiffusionText2WorldGenerationPipeline
10
+ # import .cosmos1.utils.log as log
11
+ # import .cosmos1.utils.misc as misc
12
+ # from .cosmos1.utils.io import read_prompts_from_file, save_video
13
 
14
  class DiffusionText2WorldConfig(PretrainedConfig):
15
  model_type = "DiffusionText2World"