YiYiXu
/

modular-diffdiff

Model card Files Files and versions

YiYiXu commited on Jun 19

Commit

f5d261f

·

verified ·

1 Parent(s): 0aecace

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -14,7 +14,7 @@ repo_id = "YiYiXu/modular-diffdiff"
 components = ComponentsManager()
 diffdiff_pipeline = ModularPipeline.from_pretrained(repo_id, trust_remote_code=True, component_manager=components, collection="diffdiff")
-diffdiff_pipeline.loader.load(torch_dtype=torch.float16)
 components.enable_auto_cpu_offload()

 components = ComponentsManager()
 diffdiff_pipeline = ModularPipeline.from_pretrained(repo_id, trust_remote_code=True, component_manager=components, collection="diffdiff")
+diffdiff_pipeline.load_componenets(torch_dtype=torch.float16)
 components.enable_auto_cpu_offload()