Spaces:
Running
Running
Update app.py
Browse files
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,
|