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
|