EITD commited on
Commit
dbd38f1
·
1 Parent(s): 4baf2bd

bypass cuda

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -2,7 +2,9 @@ import gradio as gr
2
  # from huggingface_hub import InferenceClient
3
  # from peft import AutoPeftModelForCausalLM
4
  # from transformers import AutoTokenizer, TextStreamer, BitsAndBytesConfig
5
- from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
 
 
6
  from unsloth import FastLanguageModel
7
  """
8
  For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 
2
  # from huggingface_hub import InferenceClient
3
  # from peft import AutoPeftModelForCausalLM
4
  # from transformers import AutoTokenizer, TextStreamer, BitsAndBytesConfig
5
+ # from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
6
+ import os
7
+ os.environ["CUDA_VISIBLE_DEVICES"] = ""
8
  from unsloth import FastLanguageModel
9
  """
10
  For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference