Commit
•
9c31218
1
Parent(s):
61ba2f1
Update README.md
Browse files
README.md
CHANGED
@@ -6,8 +6,14 @@ tags:
|
|
6 |
|
7 |
```python
|
8 |
from diffusers import DiffusionPipeline
|
|
|
9 |
|
10 |
-
|
11 |
-
|
|
|
12 |
|
13 |
-
|
|
|
|
|
|
|
|
|
|
6 |
|
7 |
```python
|
8 |
from diffusers import DiffusionPipeline
|
9 |
+
pipeline = DiffusionPipeline.from_pretrained("hf-internal-testing/tiny-sdxl-custom-components", trust_remote_code=True)
|
10 |
|
11 |
+
assert pipeline.config.unet == ('diffusers_modules.local.my_unet_model', 'MyUNetModel')
|
12 |
+
assert pipeline.config.scheduler == ('diffusers_modules.local.my_scheduler', 'MyScheduler')
|
13 |
+
assert pipeline.__class__.__name__ == "StableDiffusionXLPipeline"
|
14 |
|
15 |
+
pipeline = pipeline.to(torch_device)
|
16 |
+
images = pipeline("test", num_inference_steps=2, output_type="np")[0]
|
17 |
+
|
18 |
+
assert images.shape == (1, 64, 64, 3)
|
19 |
+
```
|