Spaces:
Runtime error
Runtime error
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 |
|