Upload folder using huggingface_hub

modeling_dream.py

@@ -559,7 +559,6 @@ class DreamPreTrainedModel(PreTrainedModel):
         token: Optional[Union[str, bool]] = None,
         revision: str = "main",
         use_safetensors: Optional[bool] = None,
-        weights_only: bool = True,
         **kwargs,
     ):
         _model = super().from_pretrained(
@@ -573,7 +572,6 @@ class DreamPreTrainedModel(PreTrainedModel):
             token=token,
             revision=revision,
             use_safetensors=use_safetensors,
-            weights_only=weights_only,
             **kwargs,
         )
         # NOTE(Lin): we need to override the generation config