FluttyProger commited on
Commit
84ad1cf
1 Parent(s): 37e3a2f

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. 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 = 'SG161222/RealVisXL_V4.0'
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
- # llm_name = 'lllyasviel/omost-dolphin-2.9-llama3-8b-4bits'
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,