prithivMLmods commited on
Commit
6cdb5fa
·
verified ·
1 Parent(s): eaab4db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -65,6 +65,8 @@ def glb_to_data_url(glb_path: str) -> str:
65
  # ---------------------------
66
  # Sambanova DeepseekR1 Clients and Chat Function
67
  # ---------------------------
 
 
68
  sambanova_client = openai.OpenAI(
69
  api_key=os.environ.get("SAMBANOVA_API_KEY"),
70
  base_url="https://api.sambanova.ai/v1",
@@ -377,7 +379,7 @@ def clean_chat_history(chat_history):
377
  # ---------------------------
378
  # Stable Diffusion XL Pipeline for Image Generation
379
  # ---------------------------
380
- MODEL_ID_SD = os.getenv("MODEL_VAL_PATH")
381
  MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
382
  USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
383
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 
65
  # ---------------------------
66
  # Sambanova DeepseekR1 Clients and Chat Function
67
  # ---------------------------
68
+
69
+ #deepseek-ai/DeepSeek-R1-Distill-Llama-70B
70
  sambanova_client = openai.OpenAI(
71
  api_key=os.environ.get("SAMBANOVA_API_KEY"),
72
  base_url="https://api.sambanova.ai/v1",
 
379
  # ---------------------------
380
  # Stable Diffusion XL Pipeline for Image Generation
381
  # ---------------------------
382
+ MODEL_ID_SD = os.getenv("MODEL_VAL_PATH") #SG161222/RealVisXL_V5.0_Lightning
383
  MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
384
  USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
385
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"