Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -430,7 +430,7 @@ def generate_30(
|
|
430 |
del processor5
|
431 |
gc.collect()
|
432 |
torch.cuda.empty_cache()
|
433 |
-
expanded = expand_prompt(
|
434 |
new_prompt = prompt+' '+expanded
|
435 |
print("-- ------------ --")
|
436 |
print("-- FINAL PROMPT --")
|
@@ -580,7 +580,7 @@ def generate_60(
|
|
580 |
del processor5
|
581 |
gc.collect()
|
582 |
torch.cuda.empty_cache()
|
583 |
-
expanded = expand_prompt(
|
584 |
new_prompt = prompt+' '+expanded
|
585 |
print("-- ------------ --")
|
586 |
print("-- FINAL PROMPT --")
|
@@ -733,7 +733,7 @@ def generate_90(
|
|
733 |
del processor5
|
734 |
gc.collect()
|
735 |
torch.cuda.empty_cache()
|
736 |
-
expanded = expand_prompt(
|
737 |
new_prompt = prompt+' '+expanded
|
738 |
print("-- ------------ --")
|
739 |
print("-- FINAL PROMPT --")
|
|
|
430 |
del processor5
|
431 |
gc.collect()
|
432 |
torch.cuda.empty_cache()
|
433 |
+
expanded = expand_prompt(captions)
|
434 |
new_prompt = prompt+' '+expanded
|
435 |
print("-- ------------ --")
|
436 |
print("-- FINAL PROMPT --")
|
|
|
580 |
del processor5
|
581 |
gc.collect()
|
582 |
torch.cuda.empty_cache()
|
583 |
+
expanded = expand_prompt(captions)
|
584 |
new_prompt = prompt+' '+expanded
|
585 |
print("-- ------------ --")
|
586 |
print("-- FINAL PROMPT --")
|
|
|
733 |
del processor5
|
734 |
gc.collect()
|
735 |
torch.cuda.empty_cache()
|
736 |
+
expanded = expand_prompt(captions)
|
737 |
new_prompt = prompt+' '+expanded
|
738 |
print("-- ------------ --")
|
739 |
print("-- FINAL PROMPT --")
|