Masrkai commited on
Commit
b380e5c
·
verified ·
1 Parent(s): 0d36818

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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 = "stabilityai/stable-zero123"
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