Spaces:
Running
on
Zero
Running
on
Zero
Update app_quant_latent1.py
Browse files- app_quant_latent1.py +4 -4
app_quant_latent1.py
CHANGED
|
@@ -317,10 +317,10 @@ def safe_get_latents(pipe, height, width, generator, device, LOGS):
|
|
| 317 |
# --------------------------
|
| 318 |
@spaces.GPU
|
| 319 |
def generate_image(prompt, height, width, steps, seed, guidance_scale=0.0):
|
| 320 |
-
LOGS = []
|
| 321 |
-
def log(msg):
|
| 322 |
-
|
| 323 |
-
|
| 324 |
|
| 325 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 326 |
generator = torch.Generator(device).manual_seed(int(seed))
|
|
|
|
| 317 |
# --------------------------
|
| 318 |
@spaces.GPU
|
| 319 |
def generate_image(prompt, height, width, steps, seed, guidance_scale=0.0):
|
| 320 |
+
LOGS = []
|
| 321 |
+
def log(msg):
|
| 322 |
+
LOGS.append(msg)
|
| 323 |
+
print(msg)
|
| 324 |
|
| 325 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 326 |
generator = torch.Generator(device).manual_seed(int(seed))
|