SaitoHoujou commited on
Commit
b5febfe
·
verified ·
1 Parent(s): 9459d0f

Update Streamlit code

Browse files
Files changed (1) hide show
  1. streamlit_app.py +2 -2
streamlit_app.py CHANGED
@@ -7,8 +7,8 @@ from controlnet_aux import HEDdetector
7
  app_controlnet_path = "SaitoHoujou/Finetuned-ControlNet_for_Diffusion-Model"
8
  app_base_model = "botp/stable-diffusion-v1-5"
9
  app_hed_model = 'lllyasviel/Annotators'
10
- app_device = "cuda" if torch.cuda.is_available() else "cpu"
11
- app_dtype = torch.float16 if app_device == "cuda" else torch.float32
12
 
13
  def generate_image(
14
  pipe,
 
7
  app_controlnet_path = "SaitoHoujou/Finetuned-ControlNet_for_Diffusion-Model"
8
  app_base_model = "botp/stable-diffusion-v1-5"
9
  app_hed_model = 'lllyasviel/Annotators'
10
+ app_device = "cpu"
11
+ app_dtype = torch.float32
12
 
13
  def generate_image(
14
  pipe,