Spaces:
Sleeping
Sleeping
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,
|