Hugging Face's logo

Spaces:
yeq6x
/
Image2Body_gradio
Running on Zero

App Files Community
yeq6x commited on
Commit
7d158d5
·
1 Parent(s): 098cbc8

to(device)

Browse files
Files changed (1) hide show
  1. scripts/process_utils.py +0 -1
scripts/process_utils.py CHANGED
@@ -84,7 +84,6 @@ def initialize_sotai_model():
84
  # Create the ControlNet pipeline
85
  sotai_gen_pipe = StableDiffusionControlNetPipeline(
86
  vae=sd_pipe.vae.to(device),
87
- torch_dtype=torch_dtype,
88
  text_encoder=sd_pipe.text_encoder,
89
  tokenizer=sd_pipe.tokenizer,
90
  unet=sd_pipe.unet.to(device),
 
84
  # Create the ControlNet pipeline
85
  sotai_gen_pipe = StableDiffusionControlNetPipeline(
86
  vae=sd_pipe.vae.to(device),
 
87
  text_encoder=sd_pipe.text_encoder,
88
  tokenizer=sd_pipe.tokenizer,
89
  unet=sd_pipe.unet.to(device),