Spaces:
Running
on
Zero
Running
on
Zero
Update
Browse files- app.py +2 -0
- tmp/config.log +1 -0
- tmp/engine_initializers.log +0 -0
- tmp/optimization.log +0 -0
app.py
CHANGED
|
@@ -38,6 +38,7 @@ def _download_from_hf() -> HFAssets:
|
|
| 38 |
repo_id=HF_REPO,
|
| 39 |
filename="weights/full_model_weights.pt",
|
| 40 |
)
|
|
|
|
| 41 |
config_path = hf_hub_download(
|
| 42 |
repo_id=HF_REPO,
|
| 43 |
filename="configs/pretrained_config.yaml",
|
|
@@ -104,6 +105,7 @@ def _get_model(device: str):
|
|
| 104 |
config_path=assets.config_path,
|
| 105 |
device=device,
|
| 106 |
verbose=False,
|
|
|
|
| 107 |
)
|
| 108 |
_cached_device = device
|
| 109 |
return _cached_ura_model
|
|
|
|
| 38 |
repo_id=HF_REPO,
|
| 39 |
filename="weights/full_model_weights.pt",
|
| 40 |
)
|
| 41 |
+
print("Weights path: ", weights_path)
|
| 42 |
config_path = hf_hub_download(
|
| 43 |
repo_id=HF_REPO,
|
| 44 |
filename="configs/pretrained_config.yaml",
|
|
|
|
| 105 |
config_path=assets.config_path,
|
| 106 |
device=device,
|
| 107 |
verbose=False,
|
| 108 |
+
skip_path_resolution=True,
|
| 109 |
)
|
| 110 |
_cached_device = device
|
| 111 |
return _cached_ura_model
|
tmp/config.log
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
2026-02-11 20:08:17,718 - utilities.config - CONFIG - CONFIG DISTRIBUTE is 'singlegpu' even though multiple GPUs are available
|
tmp/engine_initializers.log
ADDED
|
File without changes
|
tmp/optimization.log
ADDED
|
File without changes
|