Upload folder using huggingface_hub
Browse files
demo.py
CHANGED
|
@@ -46,8 +46,6 @@ model = create_model(f"tpmlp_{size}").to(device)
|
|
| 46 |
try:
|
| 47 |
load_checkpoint(model, f"../tpmlp_{size}.pth.tar", True)
|
| 48 |
except FileNotFoundError:
|
| 49 |
-
print(f"Downloading checkpoint tpmlp_{size}.")
|
| 50 |
-
os.system(f"wget https://f005.backblazeb2.com/file/cs-mixer/tpmlp_{size}.pth.tar &> /dev/null")
|
| 51 |
load_checkpoint(model, f"tpmlp_{size}.pth.tar", True)
|
| 52 |
model.eval()
|
| 53 |
|
|
|
|
| 46 |
try:
|
| 47 |
load_checkpoint(model, f"../tpmlp_{size}.pth.tar", True)
|
| 48 |
except FileNotFoundError:
|
|
|
|
|
|
|
| 49 |
load_checkpoint(model, f"tpmlp_{size}.pth.tar", True)
|
| 50 |
model.eval()
|
| 51 |
|