Update README.md
Browse files
README.md
CHANGED
@@ -16,7 +16,7 @@ How to use:
|
|
16 |
Example usage:
|
17 |
|
18 |
```python
|
19 |
-
pipeline = Zero123PlusPipeline.from_pretrained('
|
20 |
pipeline.to('cuda:0')
|
21 |
pipeline(
|
22 |
to_rgb_image(Image.open(r"condition.png"))
|
|
|
16 |
Example usage:
|
17 |
|
18 |
```python
|
19 |
+
pipeline = Zero123PlusPipeline.from_pretrained('sudo-ai/zero123plus-v1.1', torch_dtype=torch.float16)
|
20 |
pipeline.to('cuda:0')
|
21 |
pipeline(
|
22 |
to_rgb_image(Image.open(r"condition.png"))
|