alan
commited on
Commit
·
0d0c66a
1
Parent(s):
6cd713c
device
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ else:
|
|
| 27 |
torch_dtype = torch.float32
|
| 28 |
device = "cpu"
|
| 29 |
model_kwargs = {}
|
| 30 |
-
|
| 31 |
# define the pipeline
|
| 32 |
pipe = pipeline(
|
| 33 |
model=MODEL_NAME,
|
|
@@ -37,7 +37,7 @@ pipe = pipeline(
|
|
| 37 |
device=device,
|
| 38 |
model_kwargs=model_kwargs,
|
| 39 |
trust_remote_code=True
|
| 40 |
-
)
|
| 41 |
|
| 42 |
|
| 43 |
def format_time(start: Optional[float], end: Optional[float]):
|
|
|
|
| 27 |
torch_dtype = torch.float32
|
| 28 |
device = "cpu"
|
| 29 |
model_kwargs = {}
|
| 30 |
+
|
| 31 |
# define the pipeline
|
| 32 |
pipe = pipeline(
|
| 33 |
model=MODEL_NAME,
|
|
|
|
| 37 |
device=device,
|
| 38 |
model_kwargs=model_kwargs,
|
| 39 |
trust_remote_code=True
|
| 40 |
+
)
|
| 41 |
|
| 42 |
|
| 43 |
def format_time(start: Optional[float], end: Optional[float]):
|