rahul7star commited on
Commit
8ed8e09
·
verified ·
1 Parent(s): b8da832

Update app_lora1.py

Browse files
Files changed (1) hide show
  1. app_lora1.py +3 -1
app_lora1.py CHANGED
@@ -46,7 +46,8 @@ def pipeline_technology_info(pipe):
46
  # Transformer dtype
47
  if hasattr(pipe, "transformer"):
48
  try:
49
- tech.append(f"Transformer dtype: {pipe.transformer.dtype}")
 
50
  except Exception:
51
  pass
52
 
@@ -276,6 +277,7 @@ def generate_image(
276
  ):
277
  log_buffer.truncate(0)
278
  log_buffer.seek(0)
 
279
 
280
  if pipeline_name not in SCRIPT_CODE:
281
  raise RuntimeError("Pipeline not registered")
 
46
  # Transformer dtype
47
  if hasattr(pipe, "transformer"):
48
  try:
49
+ tech.append(f"Transformer full info:\n{pipe.transformer}")
50
+
51
  except Exception:
52
  pass
53
 
 
277
  ):
278
  log_buffer.truncate(0)
279
  log_buffer.seek(0)
280
+ print(prompt)
281
 
282
  if pipeline_name not in SCRIPT_CODE:
283
  raise RuntimeError("Pipeline not registered")