Ozaii commited on
Commit
23cf608
·
verified ·
1 Parent(s): f902328

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,9 +9,10 @@ from huggingface import spaces
9
  MODEL_PATH = "Ozaii/zephyr-bae"
10
  max_seq_length = 2048
11
 
12
- @spaces.GPU
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}")