ai-forever commited on
Commit
a56b7ac
·
verified ·
1 Parent(s): 74e2f0b

Update src/kandinsky.py

Browse files
Files changed (1) hide show
  1. src/kandinsky.py +4 -8
src/kandinsky.py CHANGED
@@ -35,16 +35,12 @@ MAX_SEED = np.iinfo(np.int32).max
35
  MAX_IMAGE_SIZE = 1024
36
 
37
  @spaces.GPU #[uncomment to use ZeroGPU]
38
- def generate_image(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
 
 
39
 
40
- if randomize_seed:
41
- seed = random.randint(0, MAX_SEED)
42
-
43
- generator = torch.Generator().manual_seed(seed)
44
-
45
  image = pipe(
46
  prompt = prompt,
47
- negative_prompt = negative_prompt,
48
  guidance_scale = guidance_scale,
49
  num_inference_steps = num_inference_steps,
50
  width = width,
@@ -52,7 +48,7 @@ def generate_image(prompt, negative_prompt, seed, randomize_seed, width, height,
52
  generator = generator
53
  ).images[0]
54
 
55
- return image, seed
56
 
57
 
58
  # device_map = torch.device('cuda:0')
 
35
  MAX_IMAGE_SIZE = 1024
36
 
37
  @spaces.GPU #[uncomment to use ZeroGPU]
38
+ def generate_image(prompt, width, height, guidance_scale=3, num_inference_steps=50, progress=gr.Progress(track_tqdm=True)):
39
+
40
+ # generator = torch.Generator().manual_seed(0)
41
 
 
 
 
 
 
42
  image = pipe(
43
  prompt = prompt,
 
44
  guidance_scale = guidance_scale,
45
  num_inference_steps = num_inference_steps,
46
  width = width,
 
48
  generator = generator
49
  ).images[0]
50
 
51
+ return image
52
 
53
 
54
  # device_map = torch.device('cuda:0')