Spaces:

minhdang
/

code

Paused

minhdang commited on Mar 19, 2024

Commit

92f2bfc

verified ·

1 Parent(s): abd7288

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ model_id = "deepseek-ai/deepseek-coder-33b-instruct"
 model = AutoModelForCausalLM.from_pretrained(model_id,device_map="auto",quantization_config=nf4_config)
 tokenizer=AutoTokenizer.from_pretrained(model_id)
 tokenizer.use_defaul_system_prompt=True
 @spaces.GPU
 def gen(

 model = AutoModelForCausalLM.from_pretrained(model_id,device_map="auto",quantization_config=nf4_config)
 tokenizer=AutoTokenizer.from_pretrained(model_id)
 tokenizer.use_defaul_system_prompt=True
+os.system("nvidia-smi")
 @spaces.GPU
 def gen(