Update gradio_app.py
Browse files- gradio_app.py +3 -3
gradio_app.py
CHANGED
|
@@ -35,7 +35,7 @@ import lib_omost.canvas as omost_canvas
|
|
| 35 |
|
| 36 |
# SDXL
|
| 37 |
|
| 38 |
-
sdxl_name =
|
| 39 |
# sdxl_name = 'stabilityai/stable-diffusion-xl-base-1.0'
|
| 40 |
|
| 41 |
tokenizer = CLIPTokenizer.from_pretrained(
|
|
@@ -69,8 +69,8 @@ memory_management.unload_all_models([text_encoder, text_encoder_2, vae, unet])
|
|
| 69 |
# LLM
|
| 70 |
|
| 71 |
# llm_name = 'lllyasviel/omost-phi-3-mini-128k-8bits'
|
| 72 |
-
llm_name = 'lllyasviel/omost-llama-3-8b-4bits'
|
| 73 |
-
|
| 74 |
|
| 75 |
llm_model = AutoModelForCausalLM.from_pretrained(
|
| 76 |
llm_name,
|
|
|
|
| 35 |
|
| 36 |
# SDXL
|
| 37 |
|
| 38 |
+
sdxl_name = "frankjoshua/juggernautXL_v8Rundiffusion"
|
| 39 |
# sdxl_name = 'stabilityai/stable-diffusion-xl-base-1.0'
|
| 40 |
|
| 41 |
tokenizer = CLIPTokenizer.from_pretrained(
|
|
|
|
| 69 |
# LLM
|
| 70 |
|
| 71 |
# llm_name = 'lllyasviel/omost-phi-3-mini-128k-8bits'
|
| 72 |
+
# llm_name = 'lllyasviel/omost-llama-3-8b-4bits'
|
| 73 |
+
llm_name = 'lllyasviel/omost-dolphin-2.9-llama3-8b-4bits'
|
| 74 |
|
| 75 |
llm_model = AutoModelForCausalLM.from_pretrained(
|
| 76 |
llm_name,
|