Update app.py
Browse files
app.py
CHANGED
|
@@ -48,8 +48,8 @@ UI_CONFIG = {
|
|
| 48 |
}
|
| 49 |
|
| 50 |
|
| 51 |
-
xLSTM = xLSTMWrapper(XLSTM_CONFIG).from_pretrained("AlekMan/
|
| 52 |
-
xLSTM_ft = xLSTMWrapper(XLSTM_CONFIG).from_pretrained("AlekMan/
|
| 53 |
gpt2 = GPT2LMHeadModel(GPT2_CONFIG).from_pretrained("AlekMan/HSE_AI_GPT2")
|
| 54 |
gpt2_lora = GPT2LMHeadModel(GPT2_CONFIG).from_pretrained("AlekMan/HSE_AI_GPT2")
|
| 55 |
gpt2_lora.load_adapter("AlekMan/HSE_AI_GPT2_LoRA")
|
|
|
|
| 48 |
}
|
| 49 |
|
| 50 |
|
| 51 |
+
xLSTM = xLSTMWrapper(XLSTM_CONFIG).from_pretrained("AlekMan/HSE_AI_Vanilla_XLSTM", config=XLSTM_CONFIG)
|
| 52 |
+
xLSTM_ft = xLSTMWrapper(XLSTM_CONFIG).from_pretrained("AlekMan/HSE_AI_Vanilla_XLSTM_FT", config=XLSTM_CONFIG)
|
| 53 |
gpt2 = GPT2LMHeadModel(GPT2_CONFIG).from_pretrained("AlekMan/HSE_AI_GPT2")
|
| 54 |
gpt2_lora = GPT2LMHeadModel(GPT2_CONFIG).from_pretrained("AlekMan/HSE_AI_GPT2")
|
| 55 |
gpt2_lora.load_adapter("AlekMan/HSE_AI_GPT2_LoRA")
|