turn-the-cam-anonymous commited on
Commit
51c5412
2 Parent(s): 03e871c 2311211

Merge branch 'main' of https://huggingface.co/spaces/cvlab/zero123-live into main

Browse files
Files changed (2) hide show
  1. 105000.ckpt +3 -0
  2. 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=device)
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']