modify instantiate import

model_t2w.py

@@ -26,7 +26,7 @@ from .diffusion_types import DenoisePrediction
 from .blocks import FourierFeatures
 from .pretrained_vae import BaseVAE
 from .misc import misc, Color, timer
-from . import instantiate as lazy_instantiate
+from .instantiate import instantiate as lazy_instantiate
 from .log import log
 
 

text2world_hf.py

@@ -129,7 +129,6 @@ class DiffusionText2World(PreTrainedModel):
 
     @classmethod
     def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
-        # TODO: this only builds the class, but not download the checkpoints.
         config = kwargs["config"]
         other_args = kwargs.copy()
         other_args.pop("config")