carlosabadia commited on
Commit
cbfa886
1 Parent(s): ce17b19

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -4
README.md CHANGED
@@ -36,9 +36,18 @@ Model supported in a [Gradio](https://github.com/gradio-app/gradio) Web UI and C
36
  ## Usage
37
 
38
  ```python
39
- from diffusers import StableDiffusionPipeline
 
40
 
41
- pipeline = StableDiffusionPipeline.from_pretrained('carlosabadia/hasbulla-person')
42
- image = pipeline().images[0]
43
- image
 
 
 
 
 
 
 
 
44
  ```
 
36
  ## Usage
37
 
38
  ```python
39
+ from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
40
+ import torch
41
 
42
+ model_id = "carlosabadia/hasbulla"
43
+
44
+ # Use the DPMSolverMultistepScheduler (DPM-Solver++) scheduler here instead
45
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
46
+ pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
47
+ pipe = pipe.to("cuda")
48
+
49
+ prompt = "a portrait of hasbulla person"
50
+ image = pipe(prompt).images[0]
51
+
52
+ image.save("hasbulla.png")
53
  ```