Haoxin Chen commited on
Commit
297c9c0
·
1 Parent(s): 4a341a4

remove unneeded dependencies

Browse files
Files changed (1) hide show
  1. videocrafter_test.py +3 -11
videocrafter_test.py CHANGED
@@ -1,19 +1,11 @@
1
  import os
2
- import time
3
- import yaml, math
4
- from tqdm import trange
5
- import torch
6
- import numpy as np
7
  from omegaconf import OmegaConf
8
- import torch.distributed as dist
9
- from pytorch_lightning import seed_everything
10
 
11
  from lvdm.samplers.ddim import DDIMSampler
12
- from lvdm.utils.common_utils import str2bool
13
- from lvdm.utils.saving_utils import npz_to_video_grid, npz_to_imgsheet_5d
14
  from scripts.sample_text2video import sample_text2video
15
- from scripts.sample_utils import load_model, get_conditions, make_model_input_shape, torch_to_np
16
- from lvdm.models.modules.lora import change_lora, change_lora_v2
17
 
18
  from huggingface_hub import hf_hub_download
19
 
 
1
  import os
 
 
 
 
 
2
  from omegaconf import OmegaConf
 
 
3
 
4
  from lvdm.samplers.ddim import DDIMSampler
5
+ from lvdm.utils.saving_utils import npz_to_video_grid
 
6
  from scripts.sample_text2video import sample_text2video
7
+ from scripts.sample_utils import load_model
8
+ from lvdm.models.modules.lora import change_lora_v2
9
 
10
  from huggingface_hub import hf_hub_download
11