Charles-Elena
/

ControlNet-endpoint-test

stable-diffusion

stable-diffusion-diffusers

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

charleselena commited on Jan 21

Commit

b891004

•

1 Parent(s): 4fb1dc1

change device type

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -92,7 +92,7 @@ class EndpointHandler():
  # Define Generator with seed
- self.generator = torch.Generator(device="cpu").manual_seed(3)
  def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
  """

  # Define Generator with seed
+ self.generator = torch.Generator(device=device.type).manual_seed(3)
  def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
  """