vulture990 commited on
Commit
660f05a
1 Parent(s): 7a791ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ subprocess.run(
14
  )
15
 
16
  pipeline = DiffusionPipeline.from_pretrained(
17
- "dylanebert/LGM-full",
18
- custom_pipeline="dylanebert/LGM-full",
19
  torch_dtype=torch.float16,
20
  trust_remote_code=True,
21
  ).to("cuda")
 
14
  )
15
 
16
  pipeline = DiffusionPipeline.from_pretrained(
17
+ "vulture990/2D-3D",
18
+ custom_pipeline="vulture990/2D-3D",
19
  torch_dtype=torch.float16,
20
  trust_remote_code=True,
21
  ).to("cuda")