Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
b14c4f8
1
Parent(s):
3ea8d63
update: main args
Browse files
app.py
CHANGED
|
@@ -182,6 +182,10 @@ if __name__ == '__main__':
|
|
| 182 |
args.ttf_path = 'ttf/KaiXinSongA.ttf'
|
| 183 |
args.device = 'cuda'
|
| 184 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 185 |
# load fontdiffuer pipeline
|
| 186 |
pipe = load_fontdiffuer_pipeline(args=args)
|
| 187 |
|
|
|
|
| 182 |
args.ttf_path = 'ttf/KaiXinSongA.ttf'
|
| 183 |
args.device = 'cuda'
|
| 184 |
|
| 185 |
+
args.max_batch_size = 64
|
| 186 |
+
args.num_workers = 64
|
| 187 |
+
args.adaptive_batch_size = True
|
| 188 |
+
|
| 189 |
# load fontdiffuer pipeline
|
| 190 |
pipe = load_fontdiffuer_pipeline(args=args)
|
| 191 |
|