chenyangqi commited on
Commit
5e82734
1 Parent(s): a06cc14

download in ckpt and data

Browse files
Files changed (2) hide show
  1. Dockerfile +3 -0
  2. inference_fatezero.py +1 -1
Dockerfile CHANGED
@@ -45,6 +45,9 @@ RUN pip install --no-cache-dir -U torch==1.13.1 torchvision==0.14.1
45
  COPY --chown=1000 requirements.txt /tmp/requirements.txt
46
  RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
47
 
 
 
 
48
  COPY --chown=1000 . ${HOME}/app
49
  RUN cd Tune-A-Video && patch -p1 < ../patch
50
  ENV PYTHONPATH=${HOME}/app \
 
45
  COPY --chown=1000 requirements.txt /tmp/requirements.txt
46
  RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
47
 
48
+ RUN cd FateZero/ckpt && bash download.sh
49
+ RUN cd FateZero/data && bash download.sh
50
+
51
  COPY --chown=1000 . ${HOME}/app
52
  RUN cd Tune-A-Video && patch -p1 < ../patch
53
  ENV PYTHONPATH=${HOME}/app \
inference_fatezero.py CHANGED
@@ -23,7 +23,7 @@ def merge_config_then_run(
23
  dataset_time_string = get_time_string()
24
  config_now = copy.deepcopy(Omegadict_default_edit_config)
25
  print(f"config_now['pretrained_model_path'] = model_id {model_id}")
26
- config_now['pretrained_model_path'] = model_id
27
  config_now['train_dataset']['prompt'] = source_prompt
28
  config_now['train_dataset']['path'] = data_path
29
  config_now['validation_sample_logger_config']['prompts'] = [target_prompt]
 
23
  dataset_time_string = get_time_string()
24
  config_now = copy.deepcopy(Omegadict_default_edit_config)
25
  print(f"config_now['pretrained_model_path'] = model_id {model_id}")
26
+ # config_now['pretrained_model_path'] = model_id
27
  config_now['train_dataset']['prompt'] = source_prompt
28
  config_now['train_dataset']['path'] = data_path
29
  config_now['validation_sample_logger_config']['prompts'] = [target_prompt]