dezzman commited on
Commit
90ad29a
·
verified ·
1 Parent(s): c6ed7d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -52,6 +52,9 @@ torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
52
 
53
  pipe_default = get_lora_sd_pipeline(ckpt_dir='./lora_logos', base_model_name_or_path=model_id_default, dtype=torch_dtype).to(device)
54
 
 
 
 
55
  def infer(
56
  prompt,
57
  negative_prompt,
 
52
 
53
  pipe_default = get_lora_sd_pipeline(ckpt_dir='./lora_logos', base_model_name_or_path=model_id_default, dtype=torch_dtype).to(device)
54
 
55
+ MAX_SEED = np.iinfo(np.int32).max
56
+ MAX_IMAGE_SIZE = 1024
57
+
58
  def infer(
59
  prompt,
60
  negative_prompt,