Chore: Fix Flat Ref Defaults (#21)

* Chore: Fix Flat Ref Defaults

* wrong branch

Author: painebenjamin

src/flashpack/integrations/diffusers/model.py

@@ -87,7 +87,7 @@ def from_pretrained_flashpack(
         strict: bool | None = None,
         strict_params: bool = True,
         strict_buffers: bool = False,
-        keep_flash_ref_on_model: bool = True,
+        keep_flash_ref_on_model: bool = False,
         num_streams: int = DEFAULT_NUM_STREAMS,
         chunk_bytes: int = DEFAULT_CHUNK_BYTES,
         ignore_names: list[str] | None = None,

src/flashpack/integrations/transformers/model.py

@@ -90,7 +90,7 @@ def from_pretrained_flashpack(
         strict: bool | None = None,
         strict_params: bool = True,
         strict_buffers: bool = False,
-        keep_flash_ref_on_model: bool = True,
+        keep_flash_ref_on_model: bool = False,
         num_streams: int = DEFAULT_NUM_STREAMS,
         chunk_bytes: int = DEFAULT_CHUNK_BYTES,
         ignore_names: list[str] | None = None,

src/flashpack/mixin.py

@@ -34,7 +34,7 @@ def from_flashpack(
         strict: bool | None = None,
         strict_params: bool = True,
         strict_buffers: bool = False,
-        keep_flash_ref_on_model: bool = True,
+        keep_flash_ref_on_model: bool = False,
         ignore_names: list[str] | None = None,
         ignore_prefixes: list[str] | None = None,
         ignore_suffixes: list[str] | None = None,