yeungchenwa commited on
Commit
2fc3d83
1 Parent(s): 0ae18a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -30,6 +30,7 @@ if __name__ == '__main__':
30
  args.demo = True
31
  args.ckpt_dir = 'ckpt'
32
  args.ttf_path = 'ttf/KaiXinSongA.ttf'
 
33
 
34
  # load fontdiffuer pipeline
35
  pipe = load_fontdiffuer_pipeline(args=args)
 
30
  args.demo = True
31
  args.ckpt_dir = 'ckpt'
32
  args.ttf_path = 'ttf/KaiXinSongA.ttf'
33
+ args.device = 'cpu'
34
 
35
  # load fontdiffuer pipeline
36
  pipe = load_fontdiffuer_pipeline(args=args)