LLMproj1 commited on
Commit
2293b71
·
verified ·
1 Parent(s): 0ea1eaf

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -12,7 +12,7 @@ os.system('pip install torch transformers unsloth ')
12
  os.system('pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" --no-deps xformers trl peft accelerate bitsandbytes')
13
 
14
  import gradio as gr
15
- import torch
16
  from transformers import TextStreamer
17
  from unsloth import FastLanguageModel
18
  from google.colab import drive
 
12
  os.system('pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" --no-deps xformers trl peft accelerate bitsandbytes')
13
 
14
  import gradio as gr
15
+ # import torch
16
  from transformers import TextStreamer
17
  from unsloth import FastLanguageModel
18
  from google.colab import drive