PhantHive commited on
Commit
86f7865
·
verified ·
1 Parent(s): eb0dc07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,5 @@
1
  import gradio as gr
2
- from peft import
3
- PeftModel, PeftConfig
4
  from transformers import AutoModelForCausalLM, AutoTokenizer
5
  import torch
6
 
@@ -10,6 +9,7 @@ model_id = "phearion/bigbrain-v0.0.1"
10
 
11
  # Load models and tokenizer efficiently
12
  config = PeftConfig.from_pretrained("phearion/bigbrain-v0.0.1")
 
13
  tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
14
  model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path)
15
 
 
1
  import gradio as gr
2
+ from peft import PeftModel, PeftConfig
 
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5
 
 
9
 
10
  # Load models and tokenizer efficiently
11
  config = PeftConfig.from_pretrained("phearion/bigbrain-v0.0.1")
12
+
13
  tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
14
  model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path)
15