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 |
|