Hugging Face's logo

Spaces:
chulanpro5
/
FontDiffuser-Gradio
Running on Zero

App Files Community
chulanpro5 commited on
Commit
eb1f8ea
·
1 Parent(s): 7bd89f3

update: use cuda

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +6 -6
app.py CHANGED
@@ -30,7 +30,7 @@ if __name__ == '__main__':
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)
 
30
  args.demo = True
31
  args.ckpt_dir = 'ckpt'
32
  args.ttf_path = 'ttf/KaiXinSongA.ttf'
33
+ args.device = 'cuda'
34
 
35
  # load fontdiffuer pipeline
36
  pipe = load_fontdiffuer_pipeline(args=args)
requirements.txt CHANGED
@@ -1,9 +1,9 @@
1
- torch==1.13.1
2
- torchvision==0.14.1
3
- transformers==4.33.1
4
- accelerate==0.23.0
5
- diffusers==0.22.0
6
- gradio==4.8.0
7
  pyyaml
8
  pygame
9
  opencv-python
 
1
+ torch
2
+ torchvision
3
+ transformers
4
+ accelerate
5
+ diffusers
6
+ gradio
7
  pyyaml
8
  pygame
9
  opencv-python