choco-conoz commited on
Commit
fb704f0
·
1 Parent(s): cc92ecb

feat: remove unsloth

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -2
src/streamlit_app.py CHANGED
@@ -1,5 +1,5 @@
1
  import streamlit as st
2
- from unsloth import FastLanguageModel, is_bfloat16_supported
3
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
4
 
5
  # model_id = "sentence-transformers/all-MiniLM-L6-v2"
@@ -10,7 +10,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
10
  model_id = "choco-conoz/TwinLlama-3.2-1B"
11
  tokenizer = AutoTokenizer.from_pretrained(model_id)
12
  model = AutoModelForCausalLM.from_pretrained(model_id)
13
- model = FastLanguageModel.for_inference(model)
14
 
15
  processor = pipeline(
16
  "text-generation",
 
1
  import streamlit as st
2
+ # from unsloth import FastLanguageModel, is_bfloat16_supported
3
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
4
 
5
  # model_id = "sentence-transformers/all-MiniLM-L6-v2"
 
10
  model_id = "choco-conoz/TwinLlama-3.2-1B"
11
  tokenizer = AutoTokenizer.from_pretrained(model_id)
12
  model = AutoModelForCausalLM.from_pretrained(model_id)
13
+ # model = FastLanguageModel.for_inference(model)
14
 
15
  processor = pipeline(
16
  "text-generation",