Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,9 +9,10 @@ from huggingface import spaces
|
|
9 |
MODEL_PATH = "Ozaii/zephyr-bae"
|
10 |
max_seq_length = 2048
|
11 |
|
12 |
-
|
13 |
print("Attempting to load Zephyr... Cross your fingers! 🤞")
|
14 |
try:
|
|
|
15 |
peft_config = PeftConfig.from_pretrained(MODEL_PATH)
|
16 |
base_model_name = peft_config.base_model_name_or_path
|
17 |
print(f"Loading base model: {base_model_name}")
|
|
|
9 |
MODEL_PATH = "Ozaii/zephyr-bae"
|
10 |
max_seq_length = 2048
|
11 |
|
12 |
+
|
13 |
print("Attempting to load Zephyr... Cross your fingers! 🤞")
|
14 |
try:
|
15 |
+
@spaces.GPU
|
16 |
peft_config = PeftConfig.from_pretrained(MODEL_PATH)
|
17 |
base_model_name = peft_config.base_model_name_or_path
|
18 |
print(f"Loading base model: {base_model_name}")
|