first commit

lib/util.py

@@ -25,8 +25,16 @@ import torch.nn as nn
 import numpy as np
 from einops import repeat
 
-from ..utils import instantiate_from_config
-
+#from ..utils import instantiate_from_config
+
+def instantiate_from_config(config):
+    if not "target" in config:
+        if config == '__is_first_stage__':
+            return None
+        elif config == "__is_unconditional__":
+            return None
+        raise KeyError("Expected key `target` to instantiate.")
+    return get_obj_from_str(config["target"])(**config.get("params", dict()))
 
 def make_beta_schedule(schedule, n_timestep, linear_start=1e-4, linear_end=2e-2, cosine_s=8e-3):
     if schedule == "linear":