rahul7star commited on
Commit
93c177b
·
verified ·
1 Parent(s): 8f03f67

Update app_quant_latent1.py

Browse files
Files changed (1) hide show
  1. 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
- 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))
 
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))