Thaweewat commited on
Commit
35775c4
·
1 Parent(s): 0fffd59

Clean code

Browse files
Files changed (2) hide show
  1. .idea/workspace.xml +1 -1
  2. app.py +0 -2
.idea/workspace.xml CHANGED
@@ -46,7 +46,7 @@
46
  <option name="number" value="Default" />
47
  <option name="presentableId" value="Default" />
48
  <updated>1677222707113</updated>
49
- <workItem from="1677222708286" duration="3889000" />
50
  </task>
51
  <servers />
52
  </component>
 
46
  <option name="number" value="Default" />
47
  <option name="presentableId" value="Default" />
48
  <updated>1677222707113</updated>
49
+ <workItem from="1677222708286" duration="4501000" />
50
  </task>
51
  <servers />
52
  </component>
app.py CHANGED
@@ -17,7 +17,6 @@ canny_checkpoint = "models/control_sd15_canny.pth"
17
  scribble_checkpoint = "models/control_sd15_scribble.pth"
18
  pose_checkpoint = "models/control_sd15_openpose.pth"
19
 
20
-
21
  canny_model = create_model('./models/cldm_v15.yaml').cpu()
22
  canny_model.load_state_dict(load_state_dict(cached_download(
23
  hf_hub_url(REPO_ID, canny_checkpoint)
@@ -38,7 +37,6 @@ scribble_model.load_state_dict(load_state_dict(cached_download(
38
  ), location='cpu'))
39
  scribble_model = scribble_model.cuda()
40
  ddim_sampler_scribble = DDIMSampler(scribble_model)
41
-
42
  save_memory = False
43
 
44
  def process(input_image, prompt, input_control, num_samples, image_resolution, ddim_steps, scale, seed, eta, low_threshold, high_threshold):
 
17
  scribble_checkpoint = "models/control_sd15_scribble.pth"
18
  pose_checkpoint = "models/control_sd15_openpose.pth"
19
 
 
20
  canny_model = create_model('./models/cldm_v15.yaml').cpu()
21
  canny_model.load_state_dict(load_state_dict(cached_download(
22
  hf_hub_url(REPO_ID, canny_checkpoint)
 
37
  ), location='cpu'))
38
  scribble_model = scribble_model.cuda()
39
  ddim_sampler_scribble = DDIMSampler(scribble_model)
 
40
  save_memory = False
41
 
42
  def process(input_image, prompt, input_control, num_samples, image_resolution, ddim_steps, scale, seed, eta, low_threshold, high_threshold):