EthanZyh commited on
Commit
f4c7b3e
·
1 Parent(s): 8316c81

add some comments

Browse files
Files changed (1) hide show
  1. text2world_hf.py +1 -0
text2world_hf.py CHANGED
@@ -123,6 +123,7 @@ class DiffusionText2World(PreTrainedModel):
123
 
124
  @classmethod
125
  def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
 
126
  config = kwargs["config"]
127
  model = cls(config)
128
  return model
 
123
 
124
  @classmethod
125
  def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
126
+ # TODO: this only builds the class, but not download the checkpoints.
127
  config = kwargs["config"]
128
  model = cls(config)
129
  return model