Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -360,9 +360,9 @@ import uuid
|
|
360 |
import numpy as np
|
361 |
from PIL import Image
|
362 |
|
363 |
-
import diffusers
|
364 |
-
|
365 |
-
|
366 |
#from typing import Tuple
|
367 |
#import paramiko
|
368 |
import datetime
|
@@ -372,6 +372,7 @@ from image_gen_aux import UpscaleWithModel
|
|
372 |
import time
|
373 |
import gc
|
374 |
|
|
|
375 |
|
376 |
os.putenv('HF_HUB_ENABLE_HF_TRANSFER','1')
|
377 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
|
360 |
import numpy as np
|
361 |
from PIL import Image
|
362 |
|
363 |
+
#import diffusers
|
364 |
+
from diffusers import AutoencoderKL, StableDiffusionXLPipeline
|
365 |
+
from diffusers import EulerAncestralDiscreteScheduler
|
366 |
#from typing import Tuple
|
367 |
#import paramiko
|
368 |
import datetime
|
|
|
372 |
import time
|
373 |
import gc
|
374 |
|
375 |
+
MAX_SEED = np.iinfo(np.int64).max
|
376 |
|
377 |
os.putenv('HF_HUB_ENABLE_HF_TRANSFER','1')
|
378 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|