Update app.py
Browse files
app.py
CHANGED
|
@@ -42,7 +42,6 @@ def load_model_a(model_id):
|
|
| 42 |
device_map="auto",
|
| 43 |
trust_remote_code=True,
|
| 44 |
).eval()
|
| 45 |
-
model_a.gradient_checkpointing_enable()
|
| 46 |
return gr.update(label=model_id)
|
| 47 |
|
| 48 |
def load_model_b(model_id):
|
|
@@ -66,7 +65,6 @@ def load_model_b(model_id):
|
|
| 66 |
device_map="auto",
|
| 67 |
trust_remote_code=True,
|
| 68 |
).eval()
|
| 69 |
-
model_b.gradient_checkpointing_enable()
|
| 70 |
return gr.update(label=model_id)
|
| 71 |
|
| 72 |
@spaces.GPU()
|
|
|
|
| 42 |
device_map="auto",
|
| 43 |
trust_remote_code=True,
|
| 44 |
).eval()
|
|
|
|
| 45 |
return gr.update(label=model_id)
|
| 46 |
|
| 47 |
def load_model_b(model_id):
|
|
|
|
| 65 |
device_map="auto",
|
| 66 |
trust_remote_code=True,
|
| 67 |
).eval()
|
|
|
|
| 68 |
return gr.update(label=model_id)
|
| 69 |
|
| 70 |
@spaces.GPU()
|