grpo model by default
Browse files
app.py
CHANGED
|
@@ -2,7 +2,6 @@ from dotenv import load_dotenv
|
|
| 2 |
import gradio as gr
|
| 3 |
import numpy as np
|
| 4 |
import random
|
| 5 |
-
import json
|
| 6 |
import os
|
| 7 |
import logging
|
| 8 |
import google.generativeai as genai
|
|
@@ -27,7 +26,7 @@ else:
|
|
| 27 |
logging.warning("GEMINI_API_KEY not found in environment variables. Prompt enrichment will not work.")
|
| 28 |
|
| 29 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 30 |
-
model_repo_id = "Freepik/Pikigen-test"
|
| 31 |
|
| 32 |
if torch.cuda.is_available():
|
| 33 |
torch_dtype = torch.bfloat16
|
|
|
|
| 2 |
import gradio as gr
|
| 3 |
import numpy as np
|
| 4 |
import random
|
|
|
|
| 5 |
import os
|
| 6 |
import logging
|
| 7 |
import google.generativeai as genai
|
|
|
|
| 26 |
logging.warning("GEMINI_API_KEY not found in environment variables. Prompt enrichment will not work.")
|
| 27 |
|
| 28 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 29 |
+
model_repo_id = "Freepik/Pikigen-grpo-test"
|
| 30 |
|
| 31 |
if torch.cuda.is_available():
|
| 32 |
torch_dtype = torch.bfloat16
|