Oysiyl
/

sd-lora-android-google-toy

Inference Endpoints

Model card Files Files and versions Community

Oysiyl commited on Feb 1

Commit

d751390

•

1 Parent(s): e224a9c

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -36,7 +36,7 @@ class EndpointHandler():
         if prompt is None:
             return {"error": "Please provide a prompt."}
-        generator = torch.Generator(device="cpu").manual_seed(seed)
         # hyperparamters
         num_inference_steps = data.pop("num_inference_steps", 50)

         if prompt is None:
             return {"error": "Please provide a prompt."}
+        generator = torch.Generator(device=device).manual_seed(seed)
         # hyperparamters
         num_inference_steps = data.pop("num_inference_steps", 50)