Choiszt commited on
Commit
b3e9440
·
1 Parent(s): 16eee85

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
39
 
40
  # pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-109k-release"
41
  # pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-EgoLife-Demo"
42
- pretrained = '/EgoLife-v1/EgoGPT'
43
  device = "cuda"
44
  device_map = "cuda"
45
 
 
39
 
40
  # pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-109k-release"
41
  # pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-EgoLife-Demo"
42
+ pretrained = 'EgoLife-v1/EgoGPT'
43
  device = "cuda"
44
  device_map = "cuda"
45