Commit
•
739635a
1
Parent(s):
8828c1a
Update README.md
Browse files
README.md
CHANGED
|
@@ -87,7 +87,7 @@ If you are limited by GPU memory and have less than 10GB of GPU RAM available, p
|
|
| 87 |
```py
|
| 88 |
import torch
|
| 89 |
|
| 90 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16,
|
| 91 |
pipe = pipe.to(device)
|
| 92 |
|
| 93 |
prompt = "a photo of an astronaut riding a horse on mars"
|
|
|
|
| 87 |
```py
|
| 88 |
import torch
|
| 89 |
|
| 90 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, use_auth_token=True)
|
| 91 |
pipe = pipe.to(device)
|
| 92 |
|
| 93 |
prompt = "a photo of an astronaut riding a horse on mars"
|