Mihai Chirculescu commited on
Commit ·
ed1b04d
1
Parent(s): 4c1a25c
Add schedule
Browse files- app.py +1 -2
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -5,8 +5,7 @@ from typing import Any, Iterable
|
|
| 5 |
import gradio as gr
|
| 6 |
import spaces
|
| 7 |
import torch
|
| 8 |
-
from llm_steer import Steer
|
| 9 |
-
from llm_steer.schedule import DecaySchedule
|
| 10 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 11 |
|
| 12 |
LAYER_IDX = 8
|
|
|
|
| 5 |
import gradio as gr
|
| 6 |
import spaces
|
| 7 |
import torch
|
| 8 |
+
from llm_steer import Steer, DecaySchedule
|
|
|
|
| 9 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 10 |
|
| 11 |
LAYER_IDX = 8
|
requirements.txt
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
llm_steer >= 2.0.
|
| 2 |
transformers
|
| 3 |
torch
|
| 4 |
accelerate
|
|
|
|
| 1 |
+
llm_steer >= 2.0.1
|
| 2 |
transformers
|
| 3 |
torch
|
| 4 |
accelerate
|