bk

anydoor/configs/inference.yaml

@@ -1,4 +1,4 @@
 
-pretrained_model: /workspace/train-wefadoor-master/step_300k_slim.ckpt
+pretrained_model: /workspace/train-wefadoor-master/anydoor/lightning_logs/version_1/checkpoints/epoch=1-step=2499.ckpt
 config_file: configs/anydoor.yaml
 save_memory: False

anydoor/run_inference_api_select.py

@@ -32,7 +32,7 @@ def load_model(new_model_ckpt):
     global model, ddim_sampler, current_model_ckpt
     if new_model_ckpt != current_model_ckpt:
         print(f"Loading new model: {new_model_ckpt}")
-        model.load_state_dict(load_state_dict(f'/workspace/train-wefadoor-master/step_300k_slim.ckpt', location='cuda'))
+        model.load_state_dict(load_state_dict(f'/workspace/train-wefadoor-master/anydoor/lightning_logs/version_1/checkpoints/epoch=1-step=2499.ckpt', location='cuda'))
         # model.load_state_dict(load_state_dict(f'/workspace/300k_wefa_boys_slim/lightning_logs/version_0/checkpoints/{new_model_ckpt}', location='cuda'))
         current_model_ckpt = new_model_ckpt
         print("New model loaded successfully.")

anydoor/run_train_anydoor.py

@@ -42,7 +42,7 @@ if save_memory:
 # accumulate_grad_batches=1
 
 # Configs
-resume_path = '/workspace/train-wefadoor-master/anydoor/step_300k_u.ckpt'
+resume_path = '/workspace/train-wefadoor-master/anydoor/lightning_logs/version_0/checkpoints/epoch=6-step=8749.ckpt'
 batch_size = 8
 logger_freq = 1000
 learning_rate = 1e-5

anydoor/{image_log/train/conditioning_gs-000000_e-000000_b-000000.png → step_311k.ckpt}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c83f57b2c574a824401313149f7c3c5e9a587f005b97cacfc2a9338a3ba91773
-size 170987
+oid sha256:7b90850c52742351419086745d4ebca7171ca678e25003f3bc512c74afc73429
+size 16836620107

anydoor/step_320k_P.ckpt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:27677fbf4254e0f70f37f575658156e5fa193255de025aa7a0252cc8c15748c6
-size 9808231801