SuyeonJ commited on
Commit
c93afe4
ยท
verified ยท
1 Parent(s): 8d015d4

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,8 +40,8 @@ def check_valid_login(user_name, password):
40
  MAX_FRAME = 24
41
 
42
  #VFI inference ์ฝ”๋“œ๋ฅผ ๊ทธ๋Œ€๋กœ ๊ฐ€์ ธ์™”์Šต๋‹ˆ๋‹ค.
43
- DEVICE = 0#"cuda"
44
- torch.cuda.set_device(DEVICE)
45
  #ROOT = args.root
46
  #SAVE_ROOT = f'output'
47
  SCALE = 1
 
40
  MAX_FRAME = 24
41
 
42
  #VFI inference ์ฝ”๋“œ๋ฅผ ๊ทธ๋Œ€๋กœ ๊ฐ€์ ธ์™”์Šต๋‹ˆ๋‹ค.
43
+ DEVICE = "cpu"#0#"cuda"
44
+ #torch.cuda.set_device(DEVICE)
45
  #ROOT = args.root
46
  #SAVE_ROOT = f'output'
47
  SCALE = 1