dragynir commited on
Commit
f7ffa6d
·
1 Parent(s): 784b083

script with no condition

Browse files
Files changed (1) hide show
  1. src/inference_no_condition.py +2 -2
src/inference_no_condition.py CHANGED
@@ -17,7 +17,7 @@ pipe.to("cuda")
17
  with open('../examples/prompts.json', 'r') as f:
18
  prompts_list = list(json.load(f).values())
19
 
20
- image_index = 0
21
  prompt, negative_prompt = prompts_list[image_index]
22
 
23
  images = pipe(
@@ -29,6 +29,6 @@ images = pipe(
29
  num_inference_steps=20,
30
  guidance_scale=9.0,
31
  ).images[0]
32
-
33
  plt.imshow(images)
34
  plt.show()
 
17
  with open('../examples/prompts.json', 'r') as f:
18
  prompts_list = list(json.load(f).values())
19
 
20
+ image_index = 4
21
  prompt, negative_prompt = prompts_list[image_index]
22
 
23
  images = pipe(
 
29
  num_inference_steps=20,
30
  guidance_scale=9.0,
31
  ).images[0]
32
+ images.save(fr'C:\Users\dragynir\Pictures\FASHION_IMAGES\no_condition\image_{image_index+1}.png')
33
  plt.imshow(images)
34
  plt.show()