Hugging Face's logo

Spaces:
Best-codes
/
3d-model-test
Running

App Files Community
The-Best-Codes commited on
Commit
ec8835a
·
1 Parent(s): e5acefa

Fix GPU

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from diffusers.utils import export_to_gif
5
 
6
  # Load the ShapE model
7
  ckpt_id = "openai/shap-e"
8
- pipe = ShapEPipeline.from_pretrained(ckpt_id).to("cuda")
9
 
10
  def generate_shap_e_gif(prompt):
11
  guidance_scale = 15.0
 
5
 
6
  # Load the ShapE model
7
  ckpt_id = "openai/shap-e"
8
+ pipe = ShapEPipeline.from_pretrained(ckpt_id)
9
 
10
  def generate_shap_e_gif(prompt):
11
  guidance_scale = 15.0