Jingkang commited on
Commit
8aeaa4e
·
verified ·
1 Parent(s): 4091b06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,7 +49,7 @@ from huggingface_hub import hf_hub_download
49
 
50
  # Download the model checkpoint file (large-v3.pt)
51
  ego_gpt_path = hf_hub_download(
52
- repo_id="lmms-lab/EgoGPT-0.5b-Demo",
53
  filename="speech_encoder/large-v3.pt",
54
  local_dir="./",
55
  )
@@ -65,7 +65,7 @@ except PermissionError as e:
65
 
66
 
67
 
68
- pretrained = "lmms-lab/EgoGPT-0.5b-Demo"
69
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
70
  device_map = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
71
 
 
49
 
50
  # Download the model checkpoint file (large-v3.pt)
51
  ego_gpt_path = hf_hub_download(
52
+ repo_id="lmms-lab/EgoGPT-7b-Demo",
53
  filename="speech_encoder/large-v3.pt",
54
  local_dir="./",
55
  )
 
65
 
66
 
67
 
68
+ pretrained = "lmms-lab/EgoGPT-7b-Demo"
69
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
70
  device_map = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
71