Spaces:
Build error
Build error
turn-the-cam-anonymous
commited on
Merge branch 'main' of https://huggingface.co/spaces/cvlab/zero123-live into main
Browse files- 105000.ckpt +3 -0
- app.py +1 -1
105000.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f0fee4e7c1d81c70af5a5f7bb6185fd03d75f0764f8fb9a2efaf729c69679855
|
3 |
+
size 15465973531
|
app.py
CHANGED
@@ -53,7 +53,7 @@ _ARTICLE = 'See uses.md'
|
|
53 |
|
54 |
def load_model_from_config(config, ckpt, device, verbose=False):
|
55 |
print(f'Loading model from {ckpt}')
|
56 |
-
pl_sd = torch.load(ckpt, map_location=
|
57 |
if 'global_step' in pl_sd:
|
58 |
print(f'Global Step: {pl_sd["global_step"]}')
|
59 |
sd = pl_sd['state_dict']
|
|
|
53 |
|
54 |
def load_model_from_config(config, ckpt, device, verbose=False):
|
55 |
print(f'Loading model from {ckpt}')
|
56 |
+
pl_sd = torch.load(ckpt, map_location='cpu')
|
57 |
if 'global_step' in pl_sd:
|
58 |
print(f'Global Step: {pl_sd["global_step"]}')
|
59 |
sd = pl_sd['state_dict']
|