Hugging Face's logo

ByteEd
/
3DModelGeneration_PointCloud

English
Model card Files Community
1
Thomas Male commited on
Commit
fa84497
·
verified ·
1 Parent(s): f88e92d

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -33,7 +33,7 @@ class EndpointHandler():
33
 
34
  print('creating image model...')
35
  # default - base40M. use base300M or base1B for better results
36
- self.base_image_name = 'base300M'
37
  self.base_image_model = model_from_config(MODEL_CONFIGS[self.base_image_name], device)
38
  self.base_image_model.eval()
39
  self.base_diffusion = diffusion_from_config(DIFFUSION_CONFIGS[self.base_image_name])
 
33
 
34
  print('creating image model...')
35
  # default - base40M. use base300M or base1B for better results
36
+ self.base_image_name = 'base40M'
37
  self.base_image_model = model_from_config(MODEL_CONFIGS[self.base_image_name], device)
38
  self.base_image_model.eval()
39
  self.base_diffusion = diffusion_from_config(DIFFUSION_CONFIGS[self.base_image_name])