fohy24
commited on
Commit
·
57a0cc8
1
Parent(s):
3802471
remove print()
Browse files
app.py
CHANGED
@@ -47,7 +47,6 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
47 |
hf_token = os.getenv('HF_token')
|
48 |
model_path = hf_hub_download(repo_id="samfhy/morphmarket_model", filename="model_v13_1_epoch9.pt", token=hf_token)
|
49 |
checkpoint = torch.load(model_path, map_location=device)
|
50 |
-
print(checkpoint.keys())
|
51 |
|
52 |
new_layers = nn.Sequential(
|
53 |
nn.LazyLinear(2048),
|
|
|
47 |
hf_token = os.getenv('HF_token')
|
48 |
model_path = hf_hub_download(repo_id="samfhy/morphmarket_model", filename="model_v13_1_epoch9.pt", token=hf_token)
|
49 |
checkpoint = torch.load(model_path, map_location=device)
|
|
|
50 |
|
51 |
new_layers = nn.Sequential(
|
52 |
nn.LazyLinear(2048),
|