Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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-
|
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-
|
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 |
|