Update model.py
Browse files
model.py
CHANGED
@@ -9,7 +9,7 @@ def load_pipeline():
|
|
9 |
"""
|
10 |
Load the stable-zero123 model pipeline from Hugging Face.
|
11 |
"""
|
12 |
-
ckpt_id = "
|
13 |
pipe = DiffusionPipeline.from_pretrained(ckpt_id, torch_dtype=torch.float32).to("cpu")
|
14 |
return pipe
|
15 |
|
|
|
9 |
"""
|
10 |
Load the stable-zero123 model pipeline from Hugging Face.
|
11 |
"""
|
12 |
+
ckpt_id = "dylanebert/LGM-full"
|
13 |
pipe = DiffusionPipeline.from_pretrained(ckpt_id, torch_dtype=torch.float32).to("cpu")
|
14 |
return pipe
|
15 |
|