YiYiXu
/

modular-diffdiff

Model card Files Files and versions Community

YiYiXu commited on 22 days ago

Commit

d23f26e

·

verified ·

1 Parent(s): ba73c5b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -16,7 +16,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, components_manager=components, collection="diffdiff")
 diffdiff_pipeline.loader.load(torch_dtype=torch.float16)
 components.enable_auto_cpu_offload()
 ```