Hugging Face's logo

Spaces:
kumarhans
/
LGM_space
Runtime error

App Files Community
kumarhans commited on
Commit
bacab4d
·
verified ·
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
+ "kumarhans/LGM",
18
+ custom_pipeline="kumarhans/LGM",
19
  torch_dtype=torch.float16,
20
  trust_remote_code=True,
21
  ).to("cuda")