Spaces:
Runtime error
Runtime error
Commit
·
49fe6c9
1
Parent(s):
032f281
Update app.py
Browse files
app.py
CHANGED
|
@@ -57,8 +57,8 @@ def main():
|
|
| 57 |
|
| 58 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
| 59 |
|
|
|
|
| 60 |
elapsed, pipeline = elapsed_time(load_pipeline_w_lora)
|
| 61 |
-
|
| 62 |
st.write(f"Model is loaded in {elapsed} seconds!")
|
| 63 |
|
| 64 |
pipeline = pipeline.to(device)
|
|
|
|
| 57 |
|
| 58 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
| 59 |
|
| 60 |
+
st.write("Loading models...")
|
| 61 |
elapsed, pipeline = elapsed_time(load_pipeline_w_lora)
|
|
|
|
| 62 |
st.write(f"Model is loaded in {elapsed} seconds!")
|
| 63 |
|
| 64 |
pipeline = pipeline.to(device)
|