hyoungwoncho
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -34,7 +34,7 @@ device="cuda"
|
|
34 |
pipe = pipe.to(device)
|
35 |
```
|
36 |
|
37 |
-
Sampling:
|
38 |
|
39 |
```
|
40 |
output = pipe(
|
@@ -48,6 +48,20 @@ output = pipe(
|
|
48 |
).images
|
49 |
```
|
50 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
51 |
## Parameters
|
52 |
|
53 |
pag_scale : gudiance scale of PAG (ex: 5.0)
|
|
|
34 |
pipe = pipe.to(device)
|
35 |
```
|
36 |
|
37 |
+
Sampling with PAG:
|
38 |
|
39 |
```
|
40 |
output = pipe(
|
|
|
48 |
).images
|
49 |
```
|
50 |
|
51 |
+
Sampling with PAG and CFG:
|
52 |
+
|
53 |
+
```
|
54 |
+
output = pipe(
|
55 |
+
prompts,
|
56 |
+
width=512,
|
57 |
+
height=512,
|
58 |
+
num_inference_steps=50,
|
59 |
+
guidance_scale=4.0,
|
60 |
+
pag_scale=3.0,
|
61 |
+
pag_applied_layers_index=['m0']
|
62 |
+
).images
|
63 |
+
```
|
64 |
+
|
65 |
## Parameters
|
66 |
|
67 |
pag_scale : gudiance scale of PAG (ex: 5.0)
|