multimodalart HF staff commited on
Commit
88dc9bb
1 Parent(s): c94fbf5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -26,6 +26,8 @@ scheduler = Scheduler.from_pretrained(model_path, subfolder="scheduler")
26
  pipe = Pipeline(vq_model, tokenizer=tokenizer, text_encoder=text_encoder, transformer=model, scheduler=scheduler)
27
  pipe.to(device)
28
 
 
 
29
 
30
  def generate_image(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
31
  if randomize_seed or seed == 0:
 
26
  pipe = Pipeline(vq_model, tokenizer=tokenizer, text_encoder=text_encoder, transformer=model, scheduler=scheduler)
27
  pipe.to(device)
28
 
29
+ MAX_SEED = 2**32 - 1
30
+ MAX_IMAGE_SIZE = 1024
31
 
32
  def generate_image(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
33
  if randomize_seed or seed == 0: