Commit
•
da5517b
1
Parent(s):
2d8b520
Update README.md
Browse files
README.md
CHANGED
|
@@ -40,6 +40,7 @@ from diffusers import StableDiffusionPipeline
|
|
| 40 |
|
| 41 |
# basic usage
|
| 42 |
repo_id = "lavaman131/cartoonify"
|
|
|
|
| 43 |
torch_dtype = torch.float16 if device.type in ["mps", "cuda"] else torch.float32
|
| 44 |
pipeline = StableDiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype).to(device)
|
| 45 |
image = pipeline("PROMPT GOES HERE").images[0]
|
|
|
|
| 40 |
|
| 41 |
# basic usage
|
| 42 |
repo_id = "lavaman131/cartoonify"
|
| 43 |
+
device = torch.device("cuda")
|
| 44 |
torch_dtype = torch.float16 if device.type in ["mps", "cuda"] else torch.float32
|
| 45 |
pipeline = StableDiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype).to(device)
|
| 46 |
image = pipeline("PROMPT GOES HERE").images[0]
|