message

.gitignore

+*.pyc
+__pycache__
\ No newline at end of file

hydra_node/models.py

@@ -81,10 +81,10 @@ class StableDiffusionModel(nn.Module):
         return model, model_config
 
     def load_model_from_config(self, config, ckpt, verbose=False):
-        self.config.logger.info(f"Loading model from {ckpt}")
+        print(f"Loading model from {ckpt}")
         pl_sd = torch.load(ckpt, map_location="cpu")
         if "global_step" in pl_sd:
-            self.config.logger.info(f"Global Step: {pl_sd['global_step']}")
+            print(f"Global Step: {pl_sd['global_step']}")
         sd = pl_sd["state_dict"]
         model = instantiate_from_config(config.model)
         m, u = model.load_state_dict(sd, strict=False)

run_diffusion.sh

+export DTYPE="float32"
+export AMP="1"
+export MODEL="stable-diffusion"
+export DEV="True"
+export MODEL_PATH="/home/xuser/nvme1/workspace/aero/stable/sdfinetune/checkpoints/kuru30k"
+export TRANSFORMERS_CACHE="/home/xuser/nvme1/transformer_cache"
+export SENTRY_URL="https://49ca8adcf4444f82a10eae1b3fd4182f@o846434.ingest.sentry.io/6612448"
+gunicorn main:app --workers 1 --worker-class uvicorn.workers.UvicornWorker --bind 0.0.0.0:4315