Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ save_directory = "pretrained_models/t5_ckpts/t5-v1_1-xxl"
|
|
24 |
# Modeli indir
|
25 |
st_time_t5 = time.time()
|
26 |
download_t5_model(model_id, save_directory)
|
27 |
-
print(f"T5 Download Time : {
|
28 |
|
29 |
def download_model(repo_id, model_name):
|
30 |
model_path = hf_hub_download(repo_id=repo_id, filename=model_name)
|
@@ -32,7 +32,7 @@ def download_model(repo_id, model_name):
|
|
32 |
|
33 |
import glob
|
34 |
|
35 |
-
@spaces.GPU(duration=
|
36 |
def run_model(temp_config_path, ckpt_path):
|
37 |
start_time = time.time() # Record the start time
|
38 |
cmd = [
|
@@ -58,7 +58,7 @@ def run_inference(model_name, prompt_text):
|
|
58 |
config_path = config_mapping[model_name]
|
59 |
st_time_sora = time.time()
|
60 |
ckpt_path = download_model(repo_id, model_name)
|
61 |
-
print(f"Open-Sora Download Time : {
|
62 |
|
63 |
# Save prompt_text to a temporary text file
|
64 |
prompt_file = tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode='w')
|
|
|
24 |
# Modeli indir
|
25 |
st_time_t5 = time.time()
|
26 |
download_t5_model(model_id, save_directory)
|
27 |
+
print(f"T5 Download Time : {time.time()-st_time_t5} seconds")
|
28 |
|
29 |
def download_model(repo_id, model_name):
|
30 |
model_path = hf_hub_download(repo_id=repo_id, filename=model_name)
|
|
|
32 |
|
33 |
import glob
|
34 |
|
35 |
+
@spaces.GPU(duration=500)
|
36 |
def run_model(temp_config_path, ckpt_path):
|
37 |
start_time = time.time() # Record the start time
|
38 |
cmd = [
|
|
|
58 |
config_path = config_mapping[model_name]
|
59 |
st_time_sora = time.time()
|
60 |
ckpt_path = download_model(repo_id, model_name)
|
61 |
+
print(f"Open-Sora Download Time : {time.time()-st_time_sora} seconds")
|
62 |
|
63 |
# Save prompt_text to a temporary text file
|
64 |
prompt_file = tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode='w')
|