add some comments
Browse files- 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
|