FluttyProger
commited on
Commit
•
84ad1cf
1
Parent(s):
37e3a2f
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,
|