Update README.md
Browse files
README.md
CHANGED
|
@@ -16,7 +16,7 @@ repo_id = "YiYiXu/modular-diffdiff-0704"
|
|
| 16 |
|
| 17 |
components = ComponentsManager()
|
| 18 |
|
| 19 |
-
diffdiff_pipeline = ModularPipeline.from_pretrained(repo_id, trust_remote_code=True,
|
| 20 |
diffdiff_pipeline.load_default_components(torch_dtype=torch.float16)
|
| 21 |
components.enable_auto_cpu_offload()
|
| 22 |
```
|
|
|
|
| 16 |
|
| 17 |
components = ComponentsManager()
|
| 18 |
|
| 19 |
+
diffdiff_pipeline = ModularPipeline.from_pretrained(repo_id, trust_remote_code=True, components_manager=components, collection="diffdiff")
|
| 20 |
diffdiff_pipeline.load_default_components(torch_dtype=torch.float16)
|
| 21 |
components.enable_auto_cpu_offload()
|
| 22 |
```
|