AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-10-24 20:59:27 +03:00
Code Activity

additional args

Browse Source
This commit is contained in:
sayakpaul
2023-10-27 16:33:13 +05:30
parent 0ccc58a88d
commit e152eccdb0
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docs/source/en/tutorials/custom_pipelines_components.md
View File

@@ -108,7 +108,11 @@ You can load this pipeline from the Hub by specifying `trust_remote_code=True`:
```python
from diffusers import DiffusionPipeline
reloaded_pipeline = DiffusionPipeline.from_pretrained("sayakpaul/custom_pipeline_remote_code", trust_remote_code=True)
reloaded_pipeline = DiffusionPipeline.from_pretrained(
"sayakpaul/custom_pipeline_remote_code",
torch_dtype=torch.float16,
trust_remote_code=True,
).to("cuda")
```
And then perform inference: