Hugging Face's logo

Spaces:
xywwww
/
scene_diffusion___
Runtime error

App Files Community
1
xywwww commited on
Commit
86bd5c1
·
verified ·
1 Parent(s): 524ecf2

Rename app.py to app2.py

Browse files
Files changed (1) hide show
  1. app.py → app2.py +1 -0
app.py → app2.py RENAMED
@@ -21,6 +21,7 @@ apply_canny = CannyDetector()
21
  model = create_model('./models/cldm_v21_512_latctrl_coltrans.yaml').cpu()
22
  # /checkpoints/epoch=25-step=112553.ckpt
23
  model.load_state_dict(load_state_dict('xywwww/scene_diffusion/checkpoints/epoch=25-step=112553.ckpt', location='cuda'),strict=False)
 
24
  model = model.cuda()
25
  ddim_sampler = DDIMSampler(model)
26
 
 
21
  model = create_model('./models/cldm_v21_512_latctrl_coltrans.yaml').cpu()
22
  # /checkpoints/epoch=25-step=112553.ckpt
23
  model.load_state_dict(load_state_dict('xywwww/scene_diffusion/checkpoints/epoch=25-step=112553.ckpt', location='cuda'),strict=False)
24
+ pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
25
  model = model.cuda()
26
  ddim_sampler = DDIMSampler(model)
27